CycleMix项目调试代码-报错记录

新环境中安装gco时报错：

envs/pytorch_py3.7/lib/python3.7/site-packages/torchvision/transforms/functional.py:943: UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum.
  "Argument interpolation should be of type InterpolationMode instead of int. "
Traceback (most recent call last):
  File "main.py", line 239, in <module>
    main(args)
  File "main.py", line 188, in main
    train_stats = train_one_epoch(model, criterion, dataloader_train_dict, optimizer, device, epoch,args,writer)
/CycleMix-main/engines.py", line 143, in train_one_epoch
    out, reweighted_target, indices_transport, mask_transport = mixup_process(samples_var_256, targets_onehot_256, args=args, grad= unary, noise = noise)
  File "/CycleMix-main/mixup.py", line 75, in mixup_process
    noise=noise, adv_mask1=adv_mask1, adv_mask2=adv_mask2, mp=mp, device='cuda')
  File "CycleMix-main/mixup.py", line 230, in mixup_graph
    mask.append(graphcut_multi(unary2[i], unary1[i], pw_x[i], pw_y[i], alpha, beta, eta, n_labels))
  File "CycleMix-main/mixup.py", line 125, in graphcut_multi
    labels = 1.0 - gco.cut_grid_graph(unary_cost, pairwise_cost, pw_x, pw_y, algorithm='swap')/(n_labels-1)
AttributeError: module 'gco' has no attribute 'cut_grid_graph'

原因是安装了错误的包，应该安装gco-v3.0

先全部卸载已安装的gco包、pygco包

再通过命令：pip install gco-wrapper安装gco-v3.0就成功了。