Commit 50ee5e00 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

changed print device type for debugging to allow for non-gpu/cuda

parent 697cd299
......@@ -142,7 +142,10 @@ class Solver():
print('TRAINING IS STARTING!')
print('=====================')
print('Model Name: {}'.format(self.model_name))
print('Device Type: {}'.format(torch.cuda.get_device_name(self.device)))
if torch.cuda.is_available():
print('Device Type: {}'.format(torch.cuda.get_device_name(self.device)))
else:
print('Device Type: {}'.format(self.device))
start_time = datetime.now()
print('Started At: {}'.format(start_time))
print('----------------------------------------')
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment