Update README.md
Browse files
README.md
CHANGED
|
@@ -102,10 +102,18 @@ Parameters of the fit()-Method:
|
|
| 102 |
```
|
| 103 |
{
|
| 104 |
"multi_target_strategy": "multi-output",
|
| 105 |
-
"loss_class": "CosineSimilarityLoss",
|
| 106 |
"epochs": 3,
|
| 107 |
-
"
|
| 108 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 109 |
}
|
| 110 |
```
|
| 111 |
|
|
|
|
| 102 |
```
|
| 103 |
{
|
| 104 |
"multi_target_strategy": "multi-output",
|
|
|
|
| 105 |
"epochs": 3,
|
| 106 |
+
"evaluation_steps": 0,
|
| 107 |
+
"evaluator": "NoneType",
|
| 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 |
|