Skip to content
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.

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

Will the Code be added? #1

Open
wobbba opened this issue Jul 28, 2023 · 1 comment
Open

Will the Code be added? #1

wobbba opened this issue Jul 28, 2023 · 1 comment

Comments

@wobbba
Copy link

wobbba commented Jul 28, 2023

It says that this repo contains the PyTorch implementation of the paper but there are no files here.

@Zhongying-Deng
Copy link
Owner

Hi,

Thanks for your interest in our work. I am sorry for the delay. I have uploaded a version here, but the code can be messy.

Please find the code in the zip file and install the package following the README.md.

  • The training scripts are provided in the file, such as train_digit5_ms_cutmix.sh for training on Digit-Five using FixMatch-CM. The backbone models for the FixMatch-CM can be found at dassl/modeling/backbone, such as resnet_mixstyle.py or cnn_digit5_m3sda_mixstyle.py, all with the suffix ‘_mixstyle.py’.
  • After the training with FixMatch-CM, you will obtain the first-step trained model, such as model.pth.
  • The second step BORT2 training can use the script 'train_digit5_mscm_dist_net_meta_train_retrain.sh' (similar names apply to PACS or DomainNet). Note that in the sh script, you will need to specify the checkpoint from the first-step training by MODEL.INIT_WEIGHTS model.pth. The config files and the trainer (like FixMatchMSCMDistNetMetaLearnRetrain) are specified in the script. And the trainer for BORT2 is named FixMatchMSCMDistNetMetaLearnRetrain of which the py file can be found at dassl/engine/da/fixmatch_mscm_dist_net_meta_learn_retrain.py.

Please let me know if you have any questions.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants