Recruitment-Task-3-Model / training_args.json
Piotrek-max's picture
Publish reproducible DeepWeeds recruitment baseline
301f30e verified
Raw History Blame Contribute Delete
668 Bytes
{
"run": {
"device": "cuda",
"fold_files": {
"train": "AI-MED-AGH/Recruitment-Task-3/labels/train_subset0.csv",
"validation": "AI-MED-AGH/Recruitment-Task-3/labels/val_subset0.csv"
},
"images_dir": "AI-MED-AGH/Recruitment-Task-3/original/images.zip (extracted)",
"kind": "bad",
"num_workers": 0,
"pin_memory": true
},
"training": {
"batch_size": 64,
"checkpoint_selection": "minimum_validation_loss",
"class_weights": null,
"epochs": 12,
"fold": 0,
"learning_rate": 0.001,
"loss": "CrossEntropyLoss",
"optimizer": "Adam",
"scheduler": null,
"seed": 2026
}
}