a-ragab-h-m commited on
Commit
7724504
ยท
verified ยท
1 Parent(s): 90ad217

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -1
inference.py CHANGED
@@ -52,6 +52,7 @@ model = Model(
52
  decoder_input_size=params["decoder_input_size"]
53
  )
54
  model.load_state_dict(torch.load(model_path, map_location=device))
 
55
 
56
  # --- ุชู‡ูŠุฆุฉ ุงู„ู…ู…ุซู„ ูˆุงู„ู€ NN Actor ---
57
  actor = Actor(model=model,
@@ -60,7 +61,7 @@ actor = Actor(model=model,
60
  num_neighbors_action=params['num_neighbors_action'],
61
  device=device,
62
  normalize=False)
63
- actor.model.eval()
64
 
65
  nn_actor = Actor(model=None, num_movers=1, num_neighbors_action=1, device=device)
66
  nn_actor.nearest_neighbors()
 
52
  decoder_input_size=params["decoder_input_size"]
53
  )
54
  model.load_state_dict(torch.load(model_path, map_location=device))
55
+ model.eval() # โ† ู‡ุฐุง ู‡ูˆ ุงู„ุชุนุฏูŠู„ ุงู„ู…ู‡ู…
56
 
57
  # --- ุชู‡ูŠุฆุฉ ุงู„ู…ู…ุซู„ ูˆุงู„ู€ NN Actor ---
58
  actor = Actor(model=model,
 
61
  num_neighbors_action=params['num_neighbors_action'],
62
  device=device,
63
  normalize=False)
64
+
65
 
66
  nn_actor = Actor(model=None, num_movers=1, num_neighbors_action=1, device=device)
67
  nn_actor.nearest_neighbors()