I met an error when I tried to run the code with the pretrained model.
Then I modified the basepath and the config yml, and tried the command "python main_v2.py --train-s2ag False --config ./config/multimodal_context_v2.yml".
Missing key(s) in state_dict: "module.audio_encoder.conv1.weight", "module.audio_encoder.conv1.bias", "module.audio_encoder.batch_norm1.weight", "module.audio_encoder.batch_norm1.bias", "module.audio_encoder.batch_norm1.running_mean", "module.audio_encoder.batch_norm1.running_var", "module.audio_encoder.conv2.weight", "module.audio_encoder.conv2.bias", ETC.
Unexpected key(s) in state_dict: "audio_encoder.conv1.weight", "audio_encoder.conv1.bias", "audio_encoder.batch_norm1.weight", "audio_encoder.batch_norm1.bias", "audio_encoder.batch_norm1.running_mean", "audio_encoder.batch_norm1.running_var", "audio_encoder.batch_norm1.num_batches_tracked", "audio_encoder.conv2.weight", "audio_encoder.conv2.bias", "audio_encoder.batch_norm2.weight", "audio_encoder.batch_norm2.bias", ETC.
As I noticed that the only difference between the missing keys and the unexpected keys is "module.". Could you help me fix this bug? Thanks so much!