AttributeError: 'GDEncoder_stu' object has no attribute 'iafsqueeze'

[yachour1]

https://github.com/Petrichor625/HLTP/blob/ebd72e219a6b08990284d10849385bcc1699a866/trajectory_visualization.py#L120C54-L120C70

Description: I encountered an error while trying to run the script trajectory_visualization.py. The script attempts to access an attribute 'iafsqueeze.v_mem' from the GDEncoder_stu object, but it seems that this attribute doesn't exist in the model. Here are the details of the error:

Error Traceback: File "trajectory_visualization.py", line 120, in gdEncoder.iafsqueeze.v_mem = state_dict.get('iafsqueeze.v_mem', None) File "/root/anaconda3/envs/Behavior_aware/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1729, in getattr raise AttributeError(f"'{type(self).name}' object has no attribute '{name}'") AttributeError: 'GDEncoder_stu' object has no attribute 'iafsqueeze'

What I Tried:

I tried manually inspecting the model class GDEncoder_stu but couldn't find the iafsqueeze attribute. Checked if state_dict contained iafsqueeze.v_mem, but it seems that it is not part of the model.