hgkahng / WaPIRL

Original PyTorch implementation of the paper 'Self-Supervised Representation Learning for Wafer Bin Map Defect Pattern Classification'
MIT License
20 stars 8 forks source link

There was a problem running run wapirl.py #2

Open liushidu opened 4 months ago

liushidu commented 4 months ago

Hello, I have the following problem when running run wapirl.py

usage: run_wapirl.py [-h] --gpus GPUS [GPUS ...] [--num_nodes NUM_NODES]
                     [--node_rank NODE_RANK] [--dist_url DIST_URL]
                     [--dist_backend DIST_BACKEND] [--data DATA]
                     [--input_size INPUT_SIZE] --augmentation AUGMENTATION
                     --backbone_type {resnet,vggnet,alexnet} --backbone_config
                     BACKBONE_CONFIG [--decouple_input] [--epochs EPOCHS]
                     [--batch_size BATCH_SIZE] [--num_workers NUM_WORKERS]
                     [--optimizer {sgd,adamw}] [--learning_rate LEARNING_RATE]
                     [--weight_decay WEIGHT_DECAY] [--momentum MOMENTUM]
                     [--scheduler {cosine,none}] [--warmup_steps WARMUP_STEPS]
                     [--checkpoint_root CHECKPOINT_ROOT] [--write_summary]
                     [--save_every SAVE_EVERY] [--projector_type {linear,mlp}]
                     [--projector_size PROJECTOR_SIZE]
                     [--temperature TEMPERATURE]
                     [--num_negatives NUM_NEGATIVES]
                     [--loss_weight LOSS_WEIGHT]
run_wapirl.py: error: the following arguments are required: --gpus, --augmentation, --backbone_type, --backbone_config

But I've set it up in task_config.py to run with pycharm How can I make the change? Thank you very much

liushidu commented 4 months ago

I am reproducing your paper and hope to reply as soon as possible. Thanks again!