1
edit
PyTorch: Difference between revisions
m
need to set num threads explicitly in pytorch
No edit summary |
m (need to set num threads explicitly in pytorch) |
||
| Line 182: | Line 182: | ||
args = parser.parse_args() | args = parser.parse_args() | ||
torch.set_num_threads(args.num_workers) | |||
class Net(nn.Module): | class Net(nn.Module): | ||