iamwangyabin/ESN

Missing environment.yaml file

Closed this issue · 2 comments

gqk commented

Hi Yabin,

Good work! However, I noticed that the environment.yaml file is missing from the repo. Would it be possible for you to provide the environment.yaml file? I would greatly appreciate it, as it would allow me to properly run and test the code.

iamwangyabin commented

Thank you for reminding me! I have updated the environmental file.

  • 👍1
gqk commented

I have already successfully run the code, thank you @iamwangyabin for the updation!
However, the unknown param progress_bar_refresh_rate raises an error when running the code.

ESN/trainer.py

Line 12 in d163bc6

max_epochs=args.max_epochs, progress_bar_refresh_rate=1,

Moreover, I find that you use the vit_base_patch16_224 model, while L2P and DualPrompt use the vit_base_patch16_224_in21k. As far as I know, vit_base_patch16_224 is actually the vit_base_patch16_224_in21k fine-tuned on the ImageNet-1K dataset, which may bring some performance improvement.