Update README.md
Browse files
README.md
CHANGED
|
@@ -12,9 +12,7 @@ It is simply a proof-of-concept implementation showing how to load a custom Phys
|
|
| 12 |
|
| 13 |
# Data
|
| 14 |
|
| 15 |
-
The model was trained for 100 epochs on the [fabiencasenave/Rotor37](https://huggingface.co/datasets/fabiencasenave/Rotor37) dataset
|
| 16 |
-
|
| 17 |
-
The model was trained using the following split:
|
| 18 |
|
| 19 |
```python
|
| 20 |
ds_trainval = ds["train"].train_test_split(test_size=0.2, seed=42, shuffle=True)
|
|
|
|
| 12 |
|
| 13 |
# Data
|
| 14 |
|
| 15 |
+
The model was trained for 100 epochs on the [fabiencasenave/Rotor37](https://huggingface.co/datasets/fabiencasenave/Rotor37) dataset using the following split:
|
|
|
|
|
|
|
| 16 |
|
| 17 |
```python
|
| 18 |
ds_trainval = ds["train"].train_test_split(test_size=0.2, seed=42, shuffle=True)
|