Zhonathon commited on
Commit
bba8baa
1 Parent(s): dcebe54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def start(modelName):
60
  elif modelName == 'p2-t10-g.pth':
61
  load_model_name = './models/p2-t10-g.pth'
62
  else:
63
- load_model_name = None
64
  net.actor_net.load_state_dict(torch.load(load_model_name, map_location=device1)['actor_state_dict'])
65
  net.actor_optimizer.load_state_dict(torch.load(load_model_name, map_location=device1)['actor_optimizer'])
66
  net.actor_net.eval()
 
60
  elif modelName == 'p2-t10-g.pth':
61
  load_model_name = './models/p2-t10-g.pth'
62
  else:
63
+ load_model_name = './models/p1-t10-g.pth'
64
  net.actor_net.load_state_dict(torch.load(load_model_name, map_location=device1)['actor_state_dict'])
65
  net.actor_optimizer.load_state_dict(torch.load(load_model_name, map_location=device1)['actor_optimizer'])
66
  net.actor_net.eval()