Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,8 @@ ImAg4Wheat comprises 2,500,000 million images over 2,000 wheat genotypes cultiva
|
|
| 22 |
| ViT-G/14 | 1,100 M | [FoMo4Wheat_giant.pth](https://huggingface.co/PheniX-Lab/FoMo4Wheat/blob/main/weight/FoMo4Wheat_giant.pth) |
|
| 23 |
|
| 24 |
# Training
|
| 25 |
-
|
|
|
|
| 26 |
Run FoMo4Wheat training on 6 A800-80GB nodes (48 GPUs) in a SLURM cluster environment with submitit:
|
| 27 |
```
|
| 28 |
MKL_NUM_THREADS=8 OMP_NUM_THREADS=8 python FoMo4Wheat/run/train/
|
|
|
|
| 22 |
| ViT-G/14 | 1,100 M | [FoMo4Wheat_giant.pth](https://huggingface.co/PheniX-Lab/FoMo4Wheat/blob/main/weight/FoMo4Wheat_giant.pth) |
|
| 23 |
|
| 24 |
# Training
|
| 25 |
+
**Training FoMo4Wheat on ImAg4Wheat**
|
| 26 |
+
|
| 27 |
Run FoMo4Wheat training on 6 A800-80GB nodes (48 GPUs) in a SLURM cluster environment with submitit:
|
| 28 |
```
|
| 29 |
MKL_NUM_THREADS=8 OMP_NUM_THREADS=8 python FoMo4Wheat/run/train/
|