Chey97/PB_folding

1{2    "results_dir": "/home/kihara/cgalkadu/foldingdiff/results",3    "dataset_key": "cath",4    "angles_definitions": "canonical-full-angles",5    "max_seq_len": 128,6    "min_seq_len": 40,7    "trim_strategy": "randomcrop",8    "timesteps": 1000,9    "variance_schedule": "cosine",10    "variance_scale": 1.0,11    "time_encoding": "gaussian_fourier",12    "num_hidden_layers": 12,13    "hidden_size": 384,14    "intermediate_size": 768,15    "num_heads": 12,16    "position_embedding_type": "relative_key",17    "dropout_p": 0.1,18    "decoder": "mlp",19    "gradient_clip": 1.0,20    "batch_size": 64,21    "lr": 5e-05,22    "loss": "smooth_l1",23    "use_pdist_loss": 0.0,24    "l2_norm": 0.0,25    "l1_norm": 0.0,26    "circle_reg": 0.0,27    "min_epochs": 10000,28    "max_epochs": 10000,29    "early_stop_patience": 0,30    "lr_scheduler": "LinearWarmup",31    "use_swa": false,32    "subset": null,33    "exhaustive_validation_t": false,34    "syn_noiser": "",35    "single_angle_debug": -1,36    "single_timestep_debug": false,37    "cpu_only": false,38    "ngpu": -1,39    "write_valid_preds": false,40    "dryrun": false,41    "multithread": true42}