Spaces:

a-ragab-h-m
/

vrp-shanghai-transformer

Sleeping

a-ragab-h-m commited on Jun 5

Commit

f8b0dae

verified ·

1 Parent(s): 77ccba2

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -61,7 +61,7 @@ actor = Actor(model=model,
               num_neighbors_action=params['num_neighbors_action'],
               device=device,
               normalize=False)
-actor.eval_mode()
 nn_actor = Actor(model=None, num_movers=1, num_neighbors_action=1, device=device)
 nn_actor.nearest_neighbors()

               num_neighbors_action=params['num_neighbors_action'],
               device=device,
               normalize=False)
+actor.model.eval()
 nn_actor = Actor(model=None, num_movers=1, num_neighbors_action=1, device=device)
 nn_actor.nearest_neighbors()