README.md

March 22, 2022 ยท View on GitHub

Dependencies

This code requires the following:

  • python==3.8.3
  • numpy==1.18.5
  • Pillow==7.2.0
  • scikit-learn==0.23.1
  • torch==1.7.0
  • torchvision==0.8.1

Training

python main.py --max_epoch 300 --save_name GCNet_1_1 --final_dim 8 --embedding_dim 2 --using_seed --using_pseudo