Update README.md
Browse files
README.md
CHANGED
|
@@ -107,11 +107,12 @@ print("Predicted output (logits):", logits.item(), reg2cl2[str(float(round(logit
|
|
| 107 |
+ fp16: True
|
| 108 |
|
| 109 |
|
| 110 |
-
## Evaluation
|
| 111 |
|
| 112 |
-
|
| 113 |
|
| 114 |
-
## Citation [optional]
|
| 115 |
|
| 116 |
-
|
|
|
|
|
|
|
| 117 |
|
|
|
|
|
|
| 107 |
+ fp16: True
|
| 108 |
|
| 109 |
|
|
|
|
| 110 |
|
| 111 |
+
## Evaluation on test set
|
| 112 |
|
|
|
|
| 113 |
|
| 114 |
+

|
| 115 |
+
|
| 116 |
+
## Citation [optional]
|
| 117 |
|
| 118 |
+
Please refer to this repo when using the model.
|