|
{ |
|
"model_name": "ViT-B/32", |
|
"fine_tuned": true, |
|
"tokenizer_class": "CLIPTokenizer", |
|
"pretrained_weights_path": "./pytorch_model.bin", |
|
"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" |
|
} |
|
} |
|
|