File size: 608 Bytes
5de3d77 5bd64d5 5de3d77 5bd64d5 5de3d77 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
{
"model_name": "ViT-B/32",
"fine_tuned": true,
"tokenizer_class": "CLIPTokenizer",
"optimizer": {
"type": "Adam",
"learning_rate": 1e-5,
"betas": [0.9, 0.98],
"eps": 1e-6,
"weight_decay": 0.2
},
"loss_function": "CrossEntropyLoss",
"num_epochs": 10,
"batch_size": 16,
"device": "cpu",
"training_config": {
"optimizer": {
"type": "Adam",
"learning_rate": 1e-5,
"betas": [0.9, 0.98],
"eps": 1e-6,
"weight_decay": 0.2
},
"loss_function": "CrossEntropyLoss",
"num_epochs": 10,
"batch_size": 16,
"device": "cpu"
}
}
|