A small bug

[George-Chia]

I am reading your excellent code.
I found a small bug that failing to load the imitation model.
I suggest replacing model.load_state_dict_rl(torch.load(il_weight_file)) with policy.load_state_dict_rl(torch.load(il_weight_file)) in line 138 in train.py.
Thanks!

[ChanganVR]

It looks like by default the policy saves the weights of policy.model

RelationalGraphLearning/crowd_sim/envs/policy/policy.py

Lines 39 to 40 in 8e87aa5

	def save_model(self, file):
	torch.save(self.model.state_dict(), file)

Only in model-predictive RL, it saves the weights differently. Were you trying to load the weights of model-predictive RL as initialization?