Jingqi-H / GCNet

GCNet

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