Browse Source

更新 'script.py'

master
chenwenbin 4 years ago
parent
commit
5951217130
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      script.py

+ 1
- 1
script.py View File

@ -48,7 +48,7 @@ class Model(nn.Module):
model = Model(input_size, output_size)
model = nn.DataParallel(model,device_ids=[0,1,2,3]).cuda()
model.to(device)
#model.to(device)
for data in rand_loader:
if torch.cuda.is_available():


Loading…
Cancel
Save