{ "trainer_config": { "epochs": 2, "batch_size": 4, "roberta_model_path": "Lucifer-nick/hsqcSmiles", "roberta_tokenizer_path": "Lucifer-nick/hsqcSmiles", "json_path": "C:\\Users\\Lucifer_Chen\\PycharmProjects\\HSQCNMR\\data\\HSQC_shuffle.json", "flag_use_middleware": false, "save_path": "C:\\Users\\Lucifer_Chen\\PycharmProjects\\Smart2.0\\models\\", "pre_trained_model_smiles": null, "pre_trained_model_nmr": null, "pre_trained_model_clip": null, "start_epoch": 0, "use_self_correlation": false, "frozen_smiles_model": false, "frozen_nmr_model": false, "flag_save_cpu_model": false, "num_workers": 1, "use_nmr_tokenizer": false, "use_sample_balancing": false, "nmr_model_fn": "hsqc_model_version", "loss_fn": null, "smile_use_tanh": false, "flag_use_big_class": false, "class_number": 180, "flag_get_idx": false, "feature_dim": 768, "lr": 0.0001, "weight_decay": 0.00001, "print_parm": 25, "use_dict_loss": true, "name": "clip_train_multi_gpus_ori_hpcvolta2_old.py", "location": "posix.uname_result(sysname='Linux', nodename='hpcvolta1.ai.lycc.qihoo.net', release='5.8.0-49-generic', version='#55~20.04.1-Ubuntu SMP Fri Mar 26 01:01:07 UTC 2021', machine='x86_64')" }, "model_config": { "roberta_model_path": "Lucifer-nick/hsqcSmiles", "roberta_tokenizer_path": "Lucifer-nick/hsqcSmiles", "flag_use_middleware": false, "smiles_maxlen": 300, "vocab_size": 50265, "max_position_embeddings": 514, "num_attention_heads": 12, "num_hidden_layers": 12, "type_vocab_size": 1, "nmr_model_fn": "hsqc_model_version", "frozen_smiles_model": false, "frozen_nmr_model": false, "loss_fn": null, "smile_use_tanh": false, "flag_use_big_class": false, "class_number": 180, "feature_dim": 768 }, "dataset_config": { "json_path": [ "C:\\Users\\Lucifer_Chen\\PycharmProjects\\HSQCNMR\\data\\HSQC_shuffle.json" ], "smiles_tokenizer": "Lucifer-nickLucifer-nick/hsqcSmiles", "smiles_maxlen": 300, "phase": "train", "nmr_scale": 10, "nmr_min_value": -50, "nmr_max_value": 350, "nmr_augment_prob": 0.5, "nmr_augment_range": 1.5, "use_nmr_tokenizer": false, "flag_get_idx": false } }