Update run.py

run.py

@@ -92,7 +92,11 @@ actor = Actor(model=model, num_movers=num_movers,
               device=device, normalize=False)
 actor.train_mode()
 
-baseline_model = Model(input_size=input_size, embedding_size=embedding_size)
+baseline_model = Model(
+    input_size=input_size,
+    embedding_size=embedding_size,
+    decoder_input_size=params["decoder_input_size"]
+)
 baseline_actor = Actor(model=baseline_model, num_movers=num_movers,
                         num_neighbors_encoder=num_neighbors_encoder,
                         num_neighbors_action=num_neighbors_action,