Update app.py
Browse files
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 =
|
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()
|