You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I seem to find two code errors!
In base_trainer.py
code:
if gpu_id > n_gpu:
msg = "Warning: The number of GPU's configured to use is {}, but only {} are available on this machine.".format(n_gpu_use, n_gpu)
Now I havn't implemented multi-GPU training because this model doesn't require a lot of resourecs. But I will support it to support multi-GPU training, thanks for your feekback!
I seem to find two code errors!
In base_trainer.py
code:
if gpu_id > n_gpu:
msg = "Warning: The number of GPU's configured to use is {}, but only {} are available on this machine.".format(n_gpu_use, n_gpu)
code:
list_ids = list([gpu_id])
if len(device_ids) > 1:
self.model = torch.nn.DataParallel(model, device_ids=device_ids)
The text was updated successfully, but these errors were encountered: