Update run.py

run.py

@@ -81,7 +81,11 @@ tot_google_scores = google_scores.sum().item()
 input_size = validation_dataset.model_input_length()
 
 # Models
-model = Model(input_size=input_size, embedding_size=embedding_size)
+model = Model(
+    input_size=input_size,
+    embedding_size=embedding_size,
+    decoder_input_size=params["decoder_input_size"]
+)
 actor = Actor(model=model, num_movers=num_movers,
               num_neighbors_encoder=num_neighbors_encoder,
               num_neighbors_action=num_neighbors_action,