Closed lxslxs1 closed 9 months ago
我的run.sh文件为: datapath=./data4/MVTec_ad datasets=('screw' 'pill' 'capsule' 'carpet' 'grid' 'tile' 'wood' 'zipper' 'cable' 'toothbrush' 'transistor' 'metal_nut' 'bottle' 'hazelnut' 'leather') dataset_flags=($(for dataset in "${datasets[@]}"; do echo '-d '"${dataset}"; done))
python3 main.py \ --gpu 4 \ --seed 0 \ --log_group simplenet_mvtec \ --log_project MVTecAD_Results \ --results_path results \ --run_name run \ net \ -b wideresnet50 \ -le layer2 \ -le layer3 \ --pretrain_embed_dimension 1536 \ --target_embed_dimension 1536 \ --patchsize 3 \ --meta_epochs 40 \ --embedding_size 256 \ --gan_epochs 4 \ --noise_std 0.015 \ --dsc_hidden 1024 \ --dsc_layers 2 \ --dsc_margin .5 \ --pre_proj 1 \ dataset \ --batch_size 8 \ --resize 329 \ --imagesize 288 "${dataset_flags[@]}" mvtec $datapath
python的版本一定要是3.8,带的那个click是对应的
感谢!重装环境成功了!
我的run.sh文件为: datapath=./data4/MVTec_ad datasets=('screw' 'pill' 'capsule' 'carpet' 'grid' 'tile' 'wood' 'zipper' 'cable' 'toothbrush' 'transistor' 'metal_nut' 'bottle' 'hazelnut' 'leather') dataset_flags=($(for dataset in "${datasets[@]}"; do echo '-d '"${dataset}"; done))
python3 main.py \ --gpu 4 \ --seed 0 \ --log_group simplenet_mvtec \ --log_project MVTecAD_Results \ --results_path results \ --run_name run \ net \ -b wideresnet50 \ -le layer2 \ -le layer3 \ --pretrain_embed_dimension 1536 \ --target_embed_dimension 1536 \ --patchsize 3 \ --meta_epochs 40 \ --embedding_size 256 \ --gan_epochs 4 \ --noise_std 0.015 \ --dsc_hidden 1024 \ --dsc_layers 2 \ --dsc_margin .5 \ --pre_proj 1 \ dataset \ --batch_size 8 \ --resize 329 \ --imagesize 288 "${dataset_flags[@]}" mvtec $datapath