search

gcorso / DiffDock

Implementation of DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking
https://arxiv.org/abs/2210.01776
MIT License
976 stars 238 forks source link

RuntimeError: Error(s) in loading state_dict for CGModel: Missing key(s) in state_dict: "lig_node_embedding.additional_features_embedder.weight" #211

Open rytakahashi opened 2 months ago

rytakahashi commented 2 months ago

Hi, Executing pdbbind and posebusters benchmarks with v1.1

 python -m evaluate  --config default_inference_args.yaml --data_dir ../../v1.1/DiffDock/data/posebusters_benchmark_set --tqdm --dataset posebusters --split_path data/splits/posebusters_benchmark_set_ids.txt --esm_embeddings_path ../../v1.1/DiffDock/data/posebusters_ESM.pt --chain_cutoff 10 --batch_size 10 --protein_file protein --ligand_file ligands --save_visualisation

I got following errors

module.py", line 1671, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for CGModel:
    Missing key(s) in state_dict: "lig_node_embedding.additional_features_embedder.weight", "lig_node_embedding.additional_features_embedder.bias", "rec_node_embedding.additional_features_embedder.weight", "rec_node_embedding.additional_features_embedder.bias", "rec_sigma_embedding.0.weight", "rec_sigma_embedding.0.bias", "rec_sigma_embedding.3.weight", "rec_sigma_embedding.3.bias", "conv_layers.0.tp.weight", "conv_layers.0.tp.output_mask", "conv_layers.0.fc.0.0.weight", "conv_layers.0.fc.0.0.bias", "conv_layers.0.fc.0.3.weight", "conv_layers.0.fc.0.3.bias", "conv_layers.0.fc.1.0.weight", "conv_layers.0.fc.1.0.bias", "conv_layers.0.fc.1.3.weight", "conv_layers.0.fc.1.3.bias", "conv_layers.0.fc.2.0.weight", "conv_layers.0.fc.2.0.bias", "conv_layers.0.fc.2.3.weight", "conv_layers.0.fc.2.3.bias", "conv_layers.0.fc.3.0.weight", "conv_layers.0.fc.3.0.bias", "conv_layers.0.fc.3.3.weight", "conv_layers.0.fc.3.3.bias", "conv_layers.0.batch_norm.weight", "conv_layers.0.batch_norm.bias", "conv_layers.0.batch_norm.running_mean", "conv_layers.0.batch_norm.running_var", "conv_layers.1.tp.weight", "conv_layers.1.tp.output_mask", "conv_layers.1.tp._compiled_main_left_right._w3j_1_1_1", "conv_layers.1.tp._compiled_main_left_right._w3j_1_2_1", "conv_layers.1.fc.0.0.weight", "conv_layers.1.fc.0.0.bias", "conv_layers.1.fc.0.3.weight", "conv_layers.1.fc.0.3.bias", "conv_layers.1.fc.1.0.weight", "conv_layers.1.fc.1.0.bias", "conv_layers.1.fc.1.3.weight", "conv_layers.1.fc.1.3.bias", "conv_layers.1.fc.2.0.weight", "conv_layers.1.fc.2.0.bias", "conv_layers.1.fc.2.3.weight", "conv_layers.1.fc.2.3.bias", "conv_layers.1.fc.3.0.weight", "conv_layers.1.fc.3.0.bias", "conv_layers.1.fc.3.3.weight", "conv_layers.1.fc.3.3.bias", "conv_layers.1.batch_norm.weight", "conv_layers.1.batch_norm.bias", "conv_layers.1.batch_norm.running_mean", "conv_layers.1.batch_norm.running_var", "conv_layers.2.tp.weight", "conv_layers.2.tp.output_mask", "conv_layers.2.tp._compiled_main_left_right._w3j_1_1_1", "conv_layers.2.tp._compiled_main_left_right._w3j_1_2_1", "conv_layers.2.fc.0.0.weight", "conv_layers.2.fc.0.0.bias", "conv_layers.2.fc.0.3.weight", "conv_layers.2.fc.0.3.bias", "conv_layers.2.fc.1.0.weight", "conv_layers.2.fc.1.0.bias", "conv_layers.2.fc.1.3.weight", "conv_layers.2.fc.1.3.bias", "conv_layers.2.fc.2.0.weight", "conv_layers.2.fc.2.0.bias", "conv_layers.2.fc.2.3.weight", "conv_layers.2.fc.2.3.bias", "conv_layers.2.fc.3.0.weight", "conv_layers.2.fc.3.0.bias", "conv_layers.2.fc.3.3.weight", "conv_layers.2.fc.3.3.bias", "conv_layers.2.batch_norm.weight", "conv_layers.2.batch_norm.bias", "conv_layers.2.batch_norm.running_mean", "conv_layers.2.batch_norm.running_var", "conv_layers.3.tp.weight", "conv_layers.3.tp.output_mask", "conv_layers.3.tp._compiled_main_left_right._w3j_1_1_1", "conv_layers.3.tp._compiled_main_left_right._w3j_1_2_1", "conv_layers.3.fc.0.0.weight", "conv_layers.3.fc.0.0.bias", "conv_layers.3.fc.0.3.weight", "conv_layers.3.fc.0.3.bias", "conv_layers.3.fc.1.0.weight", "conv_layers.3.fc.1.0.bias", "conv_layers.3.fc.1.3.weight", "conv_layers.3.fc.1.3.bias", "conv_layers.3.fc.2.0.weight", "conv_layers.3.fc.2.0.bias", "conv_layers.3.fc.2.3.weight", "conv_layers.3.fc.2.3.bias", "conv_layers.3.fc.3.0.weight", "conv_layers.3.fc.3.0.bias", "conv_layers.3.fc.3.3.weight", "conv_layers.3.fc.3.3.bias", "conv_layers.3.batch_norm.weight", "conv_layers.3.batch_norm.bias", "conv_layers.3.batch_norm.running_mean", "conv_layers.3.batch_norm.running_var", "conv_layers.4.tp.weight", "conv_layers.4.tp.output_mask", "conv_layers.4.tp._compiled_main_left_right._w3j_1_1_1", "conv_layers.4.tp._compiled_main_left_right._w3j_1_2_1", "conv_layers.4.fc.0.0.weight", "conv_layers.4.fc.0.0.bias", "conv_layers.4.fc.0.3.weight", "conv_layers.4.fc.0.3.bias", "conv_layers.4.fc.1.0.weight", "conv_layers.4.fc.1.0.bias", "conv_layers.4.fc.1.3.weight", "conv_layers.4.fc.1.3.bias", "conv_layers.4.fc.2.0.weight", "conv_layers.4.fc.2.0.bias", "conv_layers.4.fc.2.3.weight", "conv_layers.4.fc.2.3.bias", "conv_layers.4.fc.3.0.weight", "conv_layers.4.fc.3.0.bias", "conv_layers.4.fc.3.3.weight", "conv_layers.4.fc.3.3.bias", "conv_layers.4.batch_norm.weight", "conv_layers.4.batch_norm.bias", "conv_layers.4.batch_norm.running_mean", "conv_layers.4.batch_norm.running_var", "conv_layers.5.tp.weight", "conv_layers.5.tp.output_mask", "conv_layers.5.tp._compiled_main_left_right._w3j_1_1_1", "conv_layers.5.tp._compiled_main_left_right._w3j_1_2_1", "conv_layers.5.fc.0.0.weight", "conv_layers.5.fc.0.0.bias", "conv_layers.5.fc.0.3.weight", "conv_layers.5.fc.0.3.bias", "conv_layers.5.fc.1.0.weight", "conv_layers.5.fc.1.0.bias", "conv_layers.5.fc.1.3.weight", "conv_layers.5.fc.1.3.bias", "conv_layers.5.batch_norm.weight", "conv_layers.5.batch_norm.bias", "conv_layers.5.batch_norm.running_mean", "conv_layers.5.batch_norm.running_var". 
    Unexpected key(s) in state_dict: "lig_conv_layers.0.tp.weight", "lig_conv_layers.0.tp.output_mask", "lig_conv_layers.0.fc.0.weight", "lig_conv_layers.0.fc.0.bias", "lig_conv_layers.0.fc.3.weight", "lig_conv_layers.0.fc.3.bias", "lig_conv_layers.0.batch_norm.weight", "lig_conv_layers.0.batch_norm.bias", "lig_conv_layers.0.batch_norm.running_mean", "lig_conv_layers.0.batch_norm.running_var", "lig_conv_layers.1.tp.weight", "lig_conv_layers.1.tp.output_mask", "lig_conv_layers.1.tp._compiled_main_left_right._w3j_1_1_1", "lig_conv_layers.1.tp._compiled_main_left_right._w3j_1_2_1", "lig_conv_layers.1.fc.0.weight", "lig_conv_layers.1.fc.0.bias", "lig_conv_layers.1.fc.3.weight", "lig_conv_layers.1.fc.3.bias", "lig_conv_layers.1.batch_norm.weight", "lig_conv_layers.1.batch_norm.bias", "lig_conv_layers.1.batch_norm.running_mean", "lig_conv_layers.1.batch_norm.running_var", "lig_conv_layers.2.tp.weight", "lig_conv_layers.2.tp.output_mask", "lig_conv_layers.2.tp._compiled_main_left_right._w3j_1_1_1", "lig_conv_layers.2.tp._compiled_main_left_right._w3j_1_2_1", "lig_conv_layers.2.fc.0.weight", "lig_conv_layers.2.fc.0.bias", "lig_conv_layers.2.fc.3.weight", "lig_conv_layers.2.fc.3.bias", "lig_conv_layers.2.batch_norm.weight", "lig_conv_layers.2.batch_norm.bias", "lig_conv_layers.2.batch_norm.running_mean", "lig_conv_layers.2.batch_norm.running_var", "lig_conv_layers.3.tp.weight", "lig_conv_layers.3.tp.output_mask", "lig_conv_layers.3.tp._compiled_main_left_right._w3j_1_1_1", "lig_conv_layers.3.tp._compiled_main_left_right._w3j_1_2_1", "lig_conv_layers.3.fc.0.weight", "lig_conv_layers.3.fc.0.bias", "lig_conv_layers.3.fc.3.weight", "lig_conv_layers.3.fc.3.bias", "lig_conv_layers.3.batch_norm.weight", "lig_conv_layers.3.batch_norm.bias", "lig_conv_layers.3.batch_norm.running_mean", "lig_conv_layers.3.batch_norm.running_var", "lig_conv_layers.4.tp.weight", "lig_conv_layers.4.tp.output_mask", "lig_conv_layers.4.tp._compiled_main_left_right._w3j_1_1_1", "lig_conv_layers.4.tp._compiled_main_left_right._w3j_1_2_1", "lig_conv_layers.4.fc.0.weight", "lig_conv_layers.4.fc.0.bias", "lig_conv_layers.4.fc.3.weight", "lig_conv_layers.4.fc.3.bias", "lig_conv_layers.4.batch_norm.weight", "lig_conv_layers.4.batch_norm.bias", "lig_conv_layers.4.batch_norm.running_mean", "lig_conv_layers.4.batch_norm.running_var", "lig_conv_layers.5.tp.weight", "lig_conv_layers.5.tp.output_mask", "lig_conv_layers.5.tp._compiled_main_left_right._w3j_1_1_1", "lig_conv_layers.5.tp._compiled_main_left_right._w3j_1_2_1", "lig_conv_layers.5.fc.0.weight", "lig_conv_layers.5.fc.0.bias", "lig_conv_layers.5.fc.3.weight", "lig_conv_layers.5.fc.3.bias", "lig_conv_layers.5.batch_norm.weight", "lig_conv_layers.5.batch_norm.bias", "lig_conv_layers.5.batch_norm.running_mean", "lig_conv_layers.5.batch_norm.running_var", "rec_conv_layers.0.tp.weight", "rec_conv_layers.0.tp.output_mask", "rec_conv_layers.0.fc.0.weight", "rec_conv_layers.0.fc.0.bias", "rec_conv_layers.0.fc.3.weight", "rec_conv_layers.0.fc.3.bias", "rec_conv_layers.0.batch_norm.weight", "rec_conv_layers.0.batch_norm.bias", "rec_conv_layers.0.batch_norm.running_mean", "rec_conv_layers.0.batch_norm.running_var", "rec_conv_layers.1.tp.weight", "rec_conv_layers.1.tp.output_mask", "rec_conv_layers.1.tp._compiled_main_left_right._w3j_1_1_1", "rec_conv_layers.1.tp._compiled_main_left_right._w3j_1_2_1", "rec_conv_layers.1.fc.0.weight", "rec_conv_layers.1.fc.0.bias", "rec_conv_layers.1.fc.3.weight", "rec_conv_layers.1.fc.3.bias", "rec_conv_layers.1.batch_norm.weight", "rec_conv_layers.1.batch_norm.bias", "rec_conv_layers.1.batch_norm.running_mean", "rec_conv_layers.1.batch_norm.running_var", "rec_conv_layers.2.tp.weight", "rec_conv_layers.2.tp.output_mask", "rec_conv_layers.2.tp._compiled_main_left_right._w3j_1_1_1", "rec_conv_layers.2.tp._compiled_main_left_right._w3j_1_2_1", "rec_conv_layers.2.fc.0.weight", "rec_conv_layers.2.fc.0.bias", "rec_conv_layers.2.fc.3.weight", "rec_conv_layers.2.fc.3.bias", "rec_conv_layers.2.batch_norm.weight", "rec_conv_layers.2.batch_norm.bias", "rec_conv_layers.2.batch_norm.running_mean", "rec_conv_layers.2.batch_norm.running_var", "rec_conv_layers.3.tp.weight", "rec_conv_layers.3.tp.output_mask", "rec_conv_layers.3.tp._compiled_main_left_right._w3j_1_1_1", "rec_conv_layers.3.tp._compiled_main_left_right._w3j_1_2_1", "rec_conv_layers.3.fc.0.weight", "rec_conv_layers.3.fc.0.bias", "rec_conv_layers.3.fc.3.weight", "rec_conv_layers.3.fc.3.bias", "rec_conv_layers.3.batch_norm.weight", "rec_conv_layers.3.batch_norm.bias", "rec_conv_layers.3.batch_norm.running_mean", "rec_conv_layers.3.batch_norm.running_var", "rec_conv_layers.4.tp.weight", "rec_conv_layers.4.tp.output_mask", "rec_conv_layers.4.tp._compiled_main_left_right._w3j_1_1_1", "rec_conv_layers.4.tp._compiled_main_left_right._w3j_1_2_1", "rec_conv_layers.4.fc.0.weight", "rec_conv_layers.4.fc.0.bias", "rec_conv_layers.4.fc.3.weight", "rec_conv_layers.4.fc.3.bias", "rec_conv_layers.4.batch_norm.weight", "rec_conv_layers.4.batch_norm.bias", "rec_conv_layers.4.batch_norm.running_mean", "rec_conv_layers.4.batch_norm.running_var", "rec_conv_layers.5.tp.weight", "rec_conv_layers.5.tp.output_mask", "rec_conv_layers.5.tp._compiled_main_left_right._w3j_1_1_1", "rec_conv_layers.5.tp._compiled_main_left_right._w3j_1_2_1", "rec_conv_layers.5.fc.0.weight", "rec_conv_layers.5.fc.0.bias", "rec_conv_layers.5.fc.3.weight", "rec_conv_layers.5.fc.3.bias", "rec_conv_layers.5.batch_norm.weight", "rec_conv_layers.5.batch_norm.bias", "rec_conv_layers.5.batch_norm.running_mean", "rec_conv_layers.5.batch_norm.running_var", "lig_to_rec_conv_layers.0.tp.weight", "lig_to_rec_conv_layers.0.tp.output_mask", "lig_to_rec_conv_layers.0.fc.0.weight", "lig_to_rec_conv_layers.0.fc.0.bias", "lig_to_rec_conv_layers.0.fc.3.weight", "lig_to_rec_conv_layers.0.fc.3.bias", "lig_to_rec_conv_layers.0.batch_norm.weight", "lig_to_rec_conv_layers.0.batch_norm.bias", "lig_to_rec_conv_layers.0.batch_norm.running_mean", "lig_to_rec_conv_layers.0.batch_norm.running_var", "lig_to_rec_conv_layers.1.tp.weight", "lig_to_rec_conv_layers.1.tp.output_mask", "lig_to_rec_conv_layers.1.tp._compiled_main_left_right._w3j_1_1_1", "lig_to_rec_conv_layers.1.tp._compiled_main_left_right._w3j_1_2_1", "lig_to_rec_conv_layers.1.fc.0.weight", "lig_to_rec_conv_layers.1.fc.0.bias", "lig_to_rec_conv_layers.1.fc.3.weight", "lig_to_rec_conv_layers.1.fc.3.bias", "lig_to_rec_conv_layers.1.batch_norm.weight", "lig_to_rec_conv_layers.1.batch_norm.bias", "lig_to_rec_conv_layers.1.batch_norm.running_mean", "lig_to_rec_conv_layers.1.batch_norm.running_var", "lig_to_rec_conv_layers.2.tp.weight", "lig_to_rec_conv_layers.2.tp.output_mask", "lig_to_rec_conv_layers.2.tp._compiled_main_left_right._w3j_1_1_1", "lig_to_rec_conv_layers.2.tp._compiled_main_left_right._w3j_1_2_1", "lig_to_rec_conv_layers.2.fc.0.weight", "lig_to_rec_conv_layers.2.fc.0.bias", "lig_to_rec_conv_layers.2.fc.3.weight", "lig_to_rec_conv_layers.2.fc.3.bias", "lig_to_rec_conv_layers.2.batch_norm.weight", "lig_to_rec_conv_layers.2.batch_norm.bias", "lig_to_rec_conv_layers.2.batch_norm.running_mean", "lig_to_rec_conv_layers.2.batch_norm.running_var", "lig_to_rec_conv_layers.3.tp.weight", "lig_to_rec_conv_layers.3.tp.output_mask", "lig_to_rec_conv_layers.3.tp._compiled_main_left_right._w3j_1_1_1", "lig_to_rec_conv_layers.3.tp._compiled_main_left_right._w3j_1_2_1", "lig_to_rec_conv_layers.3.fc.0.weight", "lig_to_rec_conv_layers.3.fc.0.bias", "lig_to_rec_conv_layers.3.fc.3.weight", "lig_to_rec_conv_layers.3.fc.3.bias", "lig_to_rec_conv_layers.3.batch_norm.weight", "lig_to_rec_conv_layers.3.batch_norm.bias", "lig_to_rec_conv_layers.3.batch_norm.running_mean", "lig_to_rec_conv_layers.3.batch_norm.running_var", "lig_to_rec_conv_layers.4.tp.weight", "lig_to_rec_conv_layers.4.tp.output_mask", "lig_to_rec_conv_layers.4.tp._compiled_main_left_right._w3j_1_1_1", "lig_to_rec_conv_layers.4.tp._compiled_main_left_right._w3j_1_2_1", "lig_to_rec_conv_layers.4.fc.0.weight", "lig_to_rec_conv_layers.4.fc.0.bias", "lig_to_rec_conv_layers.4.fc.3.weight", "lig_to_rec_conv_layers.4.fc.3.bias", "lig_to_rec_conv_layers.4.batch_norm.weight", "lig_to_rec_conv_layers.4.batch_norm.bias", "lig_to_rec_conv_layers.4.batch_norm.running_mean", "lig_to_rec_conv_layers.4.batch_norm.running_var", "lig_to_rec_conv_layers.5.tp.weight", "lig_to_rec_conv_layers.5.tp.output_mask", "lig_to_rec_conv_layers.5.tp._compiled_main_left_right._w3j_1_1_1", "lig_to_rec_conv_layers.5.tp._compiled_main_left_right._w3j_1_2_1", "lig_to_rec_conv_layers.5.fc.0.weight", "lig_to_rec_conv_layers.5.fc.0.bias", "lig_to_rec_conv_layers.5.fc.3.weight", "lig_to_rec_conv_layers.5.fc.3.bias", "lig_to_rec_conv_layers.5.batch_norm.weight", "lig_to_rec_conv_layers.5.batch_norm.bias", "lig_to_rec_conv_layers.5.batch_norm.running_mean", "lig_to_rec_conv_layers.5.batch_norm.running_var", "rec_to_lig_conv_layers.0.tp.weight", "rec_to_lig_conv_layers.0.tp.output_mask", "rec_to_lig_conv_layers.0.fc.0.weight", "rec_to_lig_conv_layers.0.fc.0.bias", "rec_to_lig_conv_layers.0.fc.3.weight", "rec_to_lig_conv_layers.0.fc.3.bias", "rec_to_lig_conv_layers.0.batch_norm.weight", "rec_to_lig_conv_layers.0.batch_norm.bias", "rec_to_lig_conv_layers.0.batch_norm.running_mean", "rec_to_lig_conv_layers.0.batch_norm.running_var", "rec_to_lig_conv_layers.1.tp.weight", "rec_to_lig_conv_layers.1.tp.output_mask", "rec_to_lig_conv_layers.1.tp._compiled_main_left_right._w3j_1_1_1", "rec_to_lig_conv_layers.1.tp._compiled_main_left_right._w3j_1_2_1", "rec_to_lig_conv_layers.1.fc.0.weight", "rec_to_lig_conv_layers.1.fc.0.bias", "rec_to_lig_conv_layers.1.fc.3.weight", "rec_to_lig_conv_layers.1.fc.3.bias", "rec_to_lig_conv_layers.1.batch_norm.weight", "rec_to_lig_conv_layers.1.batch_norm.bias", "rec_to_lig_conv_layers.1.batch_norm.running_mean", "rec_to_lig_conv_layers.1.batch_norm.running_var", "rec_to_lig_conv_layers.2.tp.weight", "rec_to_lig_conv_layers.2.tp.output_mask", "rec_to_lig_conv_layers.2.tp._compiled_main_left_right._w3j_1_1_1", "rec_to_lig_conv_layers.2.tp._compiled_main_left_right._w3j_1_2_1", "rec_to_lig_conv_layers.2.fc.0.weight", "rec_to_lig_conv_layers.2.fc.0.bias", "rec_to_lig_conv_layers.2.fc.3.weight", "rec_to_lig_conv_layers.2.fc.3.bias", "rec_to_lig_conv_layers.2.batch_norm.weight", "rec_to_lig_conv_layers.2.batch_norm.bias", "rec_to_lig_conv_layers.2.batch_norm.running_mean", "rec_to_lig_conv_layers.2.batch_norm.running_var", "rec_to_lig_conv_layers.3.tp.weight", "rec_to_lig_conv_layers.3.tp.output_mask", "rec_to_lig_conv_layers.3.tp._compiled_main_left_right._w3j_1_1_1", "rec_to_lig_conv_layers.3.tp._compiled_main_left_right._w3j_1_2_1", "rec_to_lig_conv_layers.3.fc.0.weight", "rec_to_lig_conv_layers.3.fc.0.bias", "rec_to_lig_conv_layers.3.fc.3.weight", "rec_to_lig_conv_layers.3.fc.3.bias", "rec_to_lig_conv_layers.3.batch_norm.weight", "rec_to_lig_conv_layers.3.batch_norm.bias", "rec_to_lig_conv_layers.3.batch_norm.running_mean", "rec_to_lig_conv_layers.3.batch_norm.running_var", "rec_to_lig_conv_layers.4.tp.weight", "rec_to_lig_conv_layers.4.tp.output_mask", "rec_to_lig_conv_layers.4.tp._compiled_main_left_right._w3j_1_1_1", "rec_to_lig_conv_layers.4.tp._compiled_main_left_right._w3j_1_2_1", "rec_to_lig_conv_layers.4.fc.0.weight", "rec_to_lig_conv_layers.4.fc.0.bias", "rec_to_lig_conv_layers.4.fc.3.weight", "rec_to_lig_conv_layers.4.fc.3.bias", "rec_to_lig_conv_layers.4.batch_norm.weight", "rec_to_lig_conv_layers.4.batch_norm.bias", "rec_to_lig_conv_layers.4.batch_norm.running_mean", "rec_to_lig_conv_layers.4.batch_norm.running_var", "rec_to_lig_conv_layers.5.tp.weight", "rec_to_lig_conv_layers.5.tp.output_mask", "rec_to_lig_conv_layers.5.tp._compiled_main_left_right._w3j_1_1_1", "rec_to_lig_conv_layers.5.tp._compiled_main_left_right._w3j_1_2_1", "rec_to_lig_conv_layers.5.fc.0.weight", "rec_to_lig_conv_layers.5.fc.0.bias", "rec_to_lig_conv_layers.5.fc.3.weight", "rec_to_lig_conv_layers.5.fc.3.bias", "rec_to_lig_conv_layers.5.batch_norm.weight", "rec_to_lig_conv_layers.5.batch_norm.bias", "rec_to_lig_conv_layers.5.batch_norm.running_mean", "rec_to_lig_conv_layers.5.batch_norm.running_var", "lig_node_embedding.linear.weight", "lig_node_embedding.linear.bias", "rec_node_embedding.linear.weight", "rec_node_embedding.linear.bias", "rec_node_embedding.lm_embedding_layer.weight", "rec_node_embedding.lm_embedding_layer.bias". 
    size mismatch for rec_edge_embedding.0.weight: copying a param with shape torch.Size([48, 128]) from checkpoint, the shape in current model is torch.Size([48, 64]).

what could be a solution for this? Thanks,