Pass a list of torch.devices to the train functions of the architectures
#80
| Job | Run time |
|---|---|
| 3s | |
| 3s |
torch.devices to the train functions of the architectures
#80
| Job | Run time |
|---|---|
| 3s | |
| 3s |