Skip to content

Commit 8a8d654

Browse files
authored
Update README.md
1 parent 9ecd54b commit 8a8d654

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

PATH/README.md

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,14 +46,11 @@ See [installation instructions](asset/INSTALL.md).
4646
## Data
4747
See [data instructions](asset/DATA.md).
4848

49-
We also provide [a toy training dataset](https://huggingface.co/BAAI/Painter/blob/main/toy_datasets.tar), with 10 samples from each required datasets. You can put it in `$Painter_ROOT/toy_datasets` and set `DATA_PATH=toy_datasets` in `$Painter_ROOT/train_painter_vit_large.sh` for toy experiments.
49+
We also provide a small training config, with 10% samples of the whole pretraining dataset.
5050

5151
## Training
5252
Download pre-trained MAE ViT-Large model from [here](https://dl.fbaipublicfiles.com/mae/pretrain/mae_pretrain_vit_large.pth) and update `path/to/mae_pretrain_vit_large.pth` in `$Painter_ROOT/train_painter_vit_large.sh`.
5353

54-
We use 8 nodes (<code>total_bsz = 8x8x32 = 2048</code>) for training:
55-
56-
5754

5855
```bash
5956
bash train_painter_vit_large.sh

0 commit comments

Comments
 (0)