Skip to content

[Fix] Fix swav readme doc #169

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 54 commits into
base: main
Choose a base branch
from
Open

[Fix] Fix swav readme doc #169

Changes from all commits
Commits
Show all changes
54 commits
Select commit Hold shift + click to select a range
143e493
transfer_swav
shiyutang Apr 18, 2023
31921ca
valid_train
shiyutang Apr 18, 2023
cc7f630
freeze_align
shiyutang Apr 20, 2023
4d8dc6b
add_ft_swav
shiyutang Apr 21, 2023
32b94c3
add_pretrain
shiyutang Apr 21, 2023
9ca16f3
Merge branch 'main' of https://github.com/PaddlePaddle/PASSL into test
shiyutang Apr 21, 2023
1da07b1
update_pretrain
shiyutang Apr 23, 2023
9dc271a
Merge branch 'main' of https://github.com/PaddlePaddle/PASSL into test
shiyutang Apr 23, 2023
709ea4d
ready_for_semi
shiyutang Apr 23, 2023
767494f
same
shiyutang Apr 23, 2023
e77313a
Merge branch 'test' of https://github.com/shiyutang/PASSL into test
shiyutang Apr 23, 2023
0021d00
split_params
shiyutang Apr 24, 2023
57af8e9
split_params
shiyutang Apr 24, 2023
120ed43
Merge branch 'test' of https://github.com/shiyutang/PASSL into test
shiyutang Apr 24, 2023
277ab82
validate_ft
shiyutang Apr 25, 2023
5e739fa
format
shiyutang Apr 26, 2023
76056f5
add_pretrain
shiyutang Apr 28, 2023
1d69baa
valid_pretrain
shiyutang May 4, 2023
5922ab1
format
shiyutang May 4, 2023
8270a3c
fix_AttrDict_error
shiyutang May 5, 2023
4e42f8e
replace_swav_resnet
shiyutang May 5, 2023
21639c2
add_ci_readme
shiyutang May 5, 2023
6814c12
compact_lr_group
shiyutang May 5, 2023
9ed1968
Merge branch 'main' of https://github.com/PaddlePaddle/PASSL into test
shiyutang May 5, 2023
77060bf
format
shiyutang May 5, 2023
45d5273
valid_ft
shiyutang May 5, 2023
3ea3e73
backbone_config
shiyutang May 6, 2023
675e075
verified
shiyutang May 6, 2023
2c25981
fix
shiyutang May 6, 2023
df744d0
update
shiyutang May 6, 2023
948e9b9
fix
shiyutang May 6, 2023
49b7dac
fix_ci
shiyutang May 8, 2023
3cfb191
edit_accord_comment
shiyutang May 15, 2023
d9f9bf1
fix
shiyutang May 15, 2023
dc55ff6
fix_by_comment
shiyutang May 16, 2023
9ce9066
Merge branch 'main' of https://github.com/PaddlePaddle/PASSL into test
shiyutang May 16, 2023
6b1c6a5
pretrained_model
shiyutang May 16, 2023
bc3f41f
add_models
shiyutang May 17, 2023
c8175d3
CI
shiyutang May 17, 2023
510ca3d
CI
shiyutang May 17, 2023
455bf84
CI
shiyutang May 17, 2023
9473b92
fix_CI
shiyutang May 17, 2023
4201699
update_10per
shiyutang May 17, 2023
0d4292a
fix_ci
shiyutang May 17, 2023
3b0862c
ft_ips
shiyutang May 18, 2023
afba8ea
fix_by_comment
shiyutang May 18, 2023
c975a33
update
shiyutang May 18, 2023
0016737
update
shiyutang May 18, 2023
363b4a4
pretrain_fix
shiyutang May 19, 2023
a556099
update
shiyutang May 19, 2023
790500f
update
shiyutang May 19, 2023
b929bf2
update_doc
shiyutang May 29, 2023
dd977a1
Merge branch 'main' into test
shiyutang May 29, 2023
ed68110
Merge branch 'main' into test
GuoxiaWang May 29, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 8 additions & 7 deletions tasks/ssl/swav/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,8 @@ python -m paddle.distributed.launch \
--master=$PADDLE_MASTER \
--devices=$CUDA_VISIBLE_DEVICES \
passl-train \
-c ./configs/swav_resnet50_224_lp_in1k_1n8c_dp_fp16o1.yaml
-c ./configs/swav_resnet50_224_lp_in1k_1n8c_dp_fp32.yaml

```

## How to End-to-End Fine-tuning
Expand Down Expand Up @@ -96,8 +97,8 @@ To perform end-to-end fine-tuning for SwAV:
--master=$PADDLE_MASTER \
--devices=$CUDA_VISIBLE_DEVICES \
passl-train \
-c ./configs/swav_resnet50_224_ft_in1k_1n4c_dp_fp16o1.yaml
-o Global.pretrained_model=./pretrained/swav/swav_resnet50_in1k_800ep_pretrained
-c ./configs/swav_resnet50_224_ft_in1k_1n4c_dp_fp32.yaml
-o Global.pretrained_model=./pretrained/swav/swav_resnet50_in1k_800ep_bz4096_pretrained
```

## Other Configurations
Expand All @@ -109,10 +110,10 @@ We provide more directly runnable configurations, see [SwAV Configurations](./co
| Model | Phase | Dataset | Configs | GPUs | Epochs | Top1 Acc (%) | Links |
| ------------- | ----------- | ------------ | ------------------------------------------------------------ | ---------- | ------ | -------- | ------------------------------------------------------------ |
| resnet50 | pretrain | ImageNet2012 | [config](./configs/swav_resnet50_224_pt_in1k_4n32c_dp_fp16o1.yaml) | A100*N2C16 | 800 | - | [model](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_800ep_bz4096_pretrained.pdparams) \| [log](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_800ep_bz4096_pretrained.log) |
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

GPUs: A100*N4C32 ?

| resnet50 | linear probe | ImageNet2012 | [config](./configs/swav_resnet50_224_lp_in1k_4n32c_dp_fp16o1.yaml) | A100*N1C8 | 100 | 75.3 | [model](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_linearprobe.pdparams) \| [log](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_linearprobe.log) |
| resnet50 | finetune-semi10 | ImageNet2012 | [config](./configs/swav_resnet50_224_ft_in1k_1n4c_dp_fp16o1.yaml) | A100*N1C4 | 20 | 69.0 | [model](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_finetune_10percent.pdparams) \| [log](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_finetune_10percent.log) |
| resnet50 | finetune-semi10 | ImageNet2012 | [config](./configs/swav_resnet50_224_ft_in1k_1n4c_dp_fp16o1.yaml) | A100*N1C4 | 20 | 55.0 | [model](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_finetune_1percent.pdparams) \| [log](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_finetune_1percent.log) |
## Citations
| resnet50 | linear probe | ImageNet2012 | [config](./configs/swav_resnet50_224_lp_in1k_4n32c_dp_fp32.yaml) | A100*N1C8 | 100 | 75.3 | [model](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_linearprobe.pdparams) \| [log](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_linearprobe.log) |
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

swav_resnet50_224_lp_in1k_1n8c_dp_fp32.yaml

| resnet50 | finetune-semi10 | ImageNet2012 | [config](./configs/swav_resnet50_224_ft_in1k_1n4c_dp_fp32.yaml) | A100*N1C4 | 20 | 69.0 | [model](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_finetune_10percent.pdparams) \| [log](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_finetune_10percent.log) |
| resnet50 | finetune-semi10 | ImageNet2012 | [config](./configs/swav_resnet50_224_ft_in1k_1n4c_dp_fp32.yaml) | A100*N1C4 | 20 | 55.0 | [model](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_finetune_1percent.pdparams) \| [log](https://passl.bj.bcebos.com/models/swav/swav_resnet50_in1k_finetune_1percent.log) |


```bibtex
@misc{caron2021unsupervised,
Expand Down