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