Upload config.json

config.json

@@ -0,0 +1,69 @@
+{
+    "trainer_config": {
+        "epochs": 2,
+        "batch_size": 4,
+        "roberta_model_path": "/hsqcSmiles",
+        "roberta_tokenizer_path": "/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": "/hsqcSmiles",
+        "roberta_tokenizer_path": "/hsqcSmiles",
+        "flag_use_middleware": false,
+        "smiles_maxlen": 300,
+        "vocab_size": 52000,
+        "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": "/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
+    }
+}