foldingdiff_cath / training_args.json
wukevin's picture
Flatten direcotry structure
98d77b1
{"angles_definitions": "canonical-full-angles", "max_seq_len": 128, "min_seq_len": 40, "trim_strategy": "randomcrop", "zero_center": true, "timesteps": 1000, "variance_schedule": "cosine", "variance_scale": 1.0, "time_encoding": "gaussian_fourier", "num_hidden_layers": 12, "hidden_size": 384, "intermediate_size": 768, "num_heads": 12, "position_embedding_type": "relative_key", "dropout_p": 0.1, "decoder": "mlp", "gradient_clip": 1.0, "batch_size": 64, "lr": 5e-05, "loss": "smooth_l1", "l2_norm": 0.0, "l1_norm": 0.0, "circle_reg": 0.0, "min_epochs": 10000, "max_epochs": 10000, "early_stop_patience": 0, "lr_scheduler": "LinearWarmup", "use_swa": false, "subset": null, "exhaustive_validation_t": false, "syn_noiser": "", "single_angle_debug": -1, "single_timestep_debug": false, "cpu_only": false, "ngpu": -1, "write_valid_preds": false, "dryrun": true, "multithread": true}