| { | |
| "name": "TinyModel1", | |
| "version": "0.2.0", | |
| "task": "text-classification", | |
| "dataset": "ag_news", | |
| "base_model": "tinymodel1-bert-scratch", | |
| "labels": [ | |
| "World", | |
| "Sports", | |
| "Business", | |
| "Sci/Tech" | |
| ], | |
| "eval_accuracy": 0.515, | |
| "train_loss": 1.1568, | |
| "num_parameters": 1339268, | |
| "max_train_samples": 3000, | |
| "max_eval_samples": 600, | |
| "epochs": 2, | |
| "batch_size": 16, | |
| "learning_rate": 0.0001 | |
| } | |