mecha2019's picture
Update config.json
65a5ee1 verified
raw
history blame contribute delete
No virus
608 Bytes
{
"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"
}
}