Update README.md
Browse files
README.md
CHANGED
|
@@ -102,18 +102,10 @@ Parameters of the fit()-Method:
|
|
| 102 |
```
|
| 103 |
{
|
| 104 |
"multi_target_strategy": "multi-output",
|
|
|
|
| 105 |
"epochs": 3,
|
| 106 |
-
"
|
| 107 |
-
"
|
| 108 |
-
"max_grad_norm": 1,
|
| 109 |
-
"optimizer_class": "<class 'torch.optim.adamw.AdamW'>",
|
| 110 |
-
"optimizer_params": {
|
| 111 |
-
"lr": 2e-05
|
| 112 |
-
},
|
| 113 |
-
"scheduler": "WarmupLinear",
|
| 114 |
-
"steps_per_epoch": 14756,
|
| 115 |
-
"warmup_steps": 1476,
|
| 116 |
-
"weight_decay": 0.01
|
| 117 |
}
|
| 118 |
```
|
| 119 |
|
|
|
|
| 102 |
```
|
| 103 |
{
|
| 104 |
"multi_target_strategy": "multi-output",
|
| 105 |
+
"loss_class": "CosineSimilarityLoss",
|
| 106 |
"epochs": 3,
|
| 107 |
+
"batch_size": 4,
|
| 108 |
+
"num_iterations": 16,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 109 |
}
|
| 110 |
```
|
| 111 |
|