| { | |
| "architectures": [ | |
| "Model" | |
| ], | |
| "dataset": { | |
| "batch_size": 32, | |
| "correlation_strength": 0.95, | |
| "drop_last": false, | |
| "pin_memory": true | |
| }, | |
| "max_epochs": 20, | |
| "model_type": "resnet", | |
| "optimizer": { | |
| "lr": 1e-05, | |
| "weight_decay": 1.0 | |
| }, | |
| "out_features": 6, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.38.2", | |
| "use_cuda": true | |
| } | |