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