| { | |
| "data_dir": "/scratch/jtb3sud/processed_elasto_plastic/global_max/normalized/small", | |
| "seq_len": 16, | |
| "stride": 16, | |
| "num_static_feats": 2, | |
| "num_dynamic_feats": 2, | |
| "hidden_dim": 128, | |
| "num_layers": 4, | |
| "epochs": 1500, | |
| "lr": 0.0001, | |
| "weight_decay": 0.0005, | |
| "grad_clip_norm": 1.0, | |
| "scheduler": "cosine", | |
| "scheduler_step": 100, | |
| "scheduler_gamma": 0.9, | |
| "device": "cuda", | |
| "num_workers": 4, | |
| "checkpoint_dir": "/scratch/jtb3sud/meshgraphnet/elasto/run1", | |
| "val_every": 10, | |
| "save_every": 100 | |
| } |