search

dotchen / LearningByCheating

(CoRL 2019) Driving in CARLA using waypoint prediction and two-stage imitation learning
MIT License
308 stars 102 forks source link

Training Phase 0 : Missing Keys while loading Teacher Net #5

Closed raks097 closed 4 years ago

raks097 commented 4 years ago

Hi, I tried training the sensori-motor agent using the train_image_phase0.py following the steps provided in README. This is resulting a RunTime Error while loading the teacher_net ( which is a BirdViewPolicyModel and was initially trained using a ResNet-18 backbone ). The code provided uses the resnet-34 backbone for both net ( ImagePolicyModel )and teacher_net and is likely giving the the runtime error of missing keys in state_dict().

I believe the code must use ResNet-18 backbone for teacher_net and ResNet-34 for net. Am I correct or is there something I am missing ?

Thank You.

raks097 commented 4 years ago

Hi, My bad, I just realised that I was using the same log folders for the both the train_birdview and train_phase0 , essentially overwriting the teacher's config file. An addition to README instructing usage of different log folders for training different phases would be really help for beginners.

Closing the issue. Thank you .