Open Dev-HJYoo opened 2 years ago
Hello,
Some errors occurred while running test.py. I changed the code to solve the error, and it runs well, leaving a message.
I've confirmed that this code change has no effect on train.py and search.py.
The error I encountered is:
utils.py 120 line
hjyoo@neuro:~/PycharmProjects/Shapley-NAS$ python test.py --model_path 'eval-train_cifar10-20221115-131311/res18_best.pth.tar' --arch 'shapley_nas' --data '/home/hjyoo/Datasets/cifar-10-batches-py/' --cutout --auxiliary 11/16 03:10:11 PM gpu device = 0 11/16 03:10:11 PM args = Namespace(data='/home/hjyoo/Datasets/cifar-10-batches-py/', batch_size=96, report_freq=50, gpu=0, init_channels=36, layers=20, model_path='eval-train_cifar10-20221115-131311/res18_best.pth.tar', auxiliary=True, cutout=True, cutout_length=16, drop_path_prob=0.2, seed=0, arch='shapley_nas') 108 108 36 108 144 36 144 144 36 144 144 36 144 144 36 144 144 36 144 144 72 144 288 72 288 288 72 288 288 72 288 288 72 288 288 72 288 288 72 288 288 144 288 576 144 576 576 144 576 576 144 576 576 144 576 576 144 576 576 144 dict_keys(['epoch', 'state_dict', 'best_acc', 'optimizer']) Traceback (most recent call last): File "/home/hjyoo/PycharmProjects/Shapley-NAS/test.py", line 102, in <module> main() File "/home/hjyoo/PycharmProjects/Shapley-NAS/test.py", line 58, in main utils.load(model, args.model_path) File "/home/hjyoo/PycharmProjects/Shapley-NAS/utils.py", line 103, in load model.load_state_dict(torch.load(model_path)) File "/home/hjyoo/anaconda3/envs/Shapley-NAS/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1667, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for NetworkCIFAR: Missing key(s) in state_dict: "stem.0.weight", "stem.1.weight", "stem.1.bias", "stem.1.running_mean", "stem.1.running_var", "cells.0.preprocess0.op.1.weight", "cells.0.preprocess0.op.2.weight", "cells.0.preprocess0.op.2.bias", "cells.0.preprocess0.op.2.running_mean", "cells.0.preprocess0.op.2.running_var", "cells.0.preprocess1.op.1.weight", "cells.0.preprocess1.op.2.weight", "cells.0.preprocess1.op.2.bias", "cells.0.preprocess1.op.2.running_mean", "cells.0.preprocess1.op.2.running_var", "cells.0._ops.0.op.1.weight", "cells.0._ops.0.op.2.weight", "cells.0._ops.0.op.3.weight", "cells.0._ops.0.op.3.bias", "cells.0._ops.0.op.3.running_mean", "cells.0._ops.0.op.3.running_var", "cells.0._ops.0.op.5.weight", "cells.0._ops.0.op.6.weight", "cells.0._ops.0.op.7.weight", "cells.0._ops.0.op.7.bias", "cells.0._ops.0.op.7.running_mean", "cells.0._ops.0.op.7.running_var", "cells.0._ops.1.op.1.weight", "cells.0._ops.1.op.2.weight", "cells.0._ops.1.op.3.weight", "cells.0._ops.1.op.3.bias", "cells.0._ops.1.op.3.running_mean", "cells.0._ops.1.op.3.running_var", "cells.0._ops.1.op.5.weight", "cells.0._ops.1.op.6.weight", "cells.0._ops.1.op.7.weight", "cells.0._ops.1.op.7.bias", "cells.0._ops.1.op.7.running_mean", "cells.0._ops.1.op.7.running_var", "cells.0._ops.2.op.1.weight", "cells.0._ops.2.op.2.weight", "cells.0._ops.2.op.3.weight", "cells.0._ops.2.op.3.bias", "cells.0._ops.2.op.3.running_mean", "cells.0._ops.2.op.3.running_var", "cells.0._ops.2.op.5.weight", "cells.0._ops.2.op.6.weight", "cells.0._ops.2.op.7.weight", "cells.0._ops.2.op.7.bias", "cells.0._ops.2.op.7.running_mean", "cells.0._ops.2.op.7.running_var", "cells.0._ops.3.op.1.weight", "cells.0._ops.3.op.2.weight", "cells.0._ops.3.op.3.weight", "cells.0._ops.3.op.3.bias", "cells.0._ops.3.op.3.running_mean", "cells.0._ops.3.op.3.running_var", "cells.0._ops.4.op.1.weight", "cells.0._ops.4.op.2.weight", "cells.0._ops.4.op.3.weight", "cells.0._ops.4.op.3.bias", "cells.0._ops.4.op.3.running_mean", "cells.0._ops.4.op.3.running_var", "cells.0._ops.6.op.1.weight", "cells.0._ops.6.op.2.weight", "cells.0._ops.6.op.3.weight", "cells.0._ops.6.op.3.bias", "cells.0._ops.6.op.3.running_mean", "cells.0._ops.6.op.3.running_var", "cells.0._ops.6.op.5.weight", "cells.0._ops.6.op.6.weight", "cells.0._ops.6.op.7.weight", "cells.0._ops.6.op.7.bias", "cells.0._ops.6.op.7.running_mean", "cells.0._ops.6.op.7.running_var", "cells.0._ops.7.op.1.weight", "cells.0._ops.7.op.2.weight", "cells.0._ops.7.op.3.weight", "cells.0._ops.7.op.3.bias", "cells.0._ops.7.op.3.running_mean", "cells.0._ops.7.op.3.running_var", "cells.1.preprocess0.op.1.weight", "cells.1.preprocess0.op.2.weight", "cells.1.preprocess0.op.2.bias", "cells.1.preprocess0.op.2.running_mean", "cells.1.preprocess0.op.2.running_var", "cells.1.preprocess1.op.1.weight", "cells.1.preprocess1.op.2.weight", "cells.1.preprocess1.op.2.bias", "cells.1.preprocess1.op.2.running_mean", "cells.1.preprocess1.op.2.running_var", "cells.1._ops.0.op.1.weight", "cells.1._ops.0.op.2.weight", "cells.1._ops.0.op.3.weight", "cells.1._ops.0.op.3.bias", "cells.1._ops.0.op.3.running_mean", "cells.1._ops.0.op.3.running_var", "cells.1._ops.0.op.5.weight", "cells.1._ops.0.op.6.weight", "cells.1._ops.0.op.7.weight", "cells.1._ops.0.op.7.bias", "cells.1._ops.0.op.7.running_mean", "cells.1._ops.0.op.7.running_var", "cells.1._ops.1.op.1.weight", "cells.1._ops.1.op.2.weight", "cells.1._ops.1.op.3.weight", "cells.1._ops.1.op.3.bias", "cells.1._ops.1.op.3.running_mean", "cells.1._ops.1.op.3.running_var", "cells.1._ops.1.op.5.weight", "cells.1._ops.1.op.6.weight", "cells.1._ops.1.op.7.weight", "cells.1._ops.1.op.7.bias", "cells.1._ops.1.op.7.running_mean", "cells.1._ops.1.op.7.running_var", "cells.1._ops.2.op.1.weight", "cells.1._ops.2.op.2.weight", "cells.1._ops.2.op.3.weight", "cells.1._ops.2.op.3.bias", "cells.1._ops.2.op.3.running_mean", "cells.1._ops.2.op.3.running_var", "cells.1._ops.2.op.5.weight", "cells.1._ops.2.op.6.weight", "cells.1._ops.2.op.7.weight", "cells.1._ops.2.op.7.bias", "cells.1._ops.2.op.7.running_mean", "cells.1._ops.2.op.7.running_var", "cells.1._ops.3.op.1.weight", "cells.1._ops.3.op.2.weight", "cells.1._ops.3.op.3.weight", "cells.1._ops.3.op.3.bias", "cells.1._ops.3.op.3.running_mean", "cells.1._ops.3.op.3.running_var", "cells.1._ops.4.op.1.weight", "cells.1._ops.4.op.2.weight", "cells.1._ops.4.op.3.weight", "cells.1._ops.4.op.3.bias", "cells.1._ops.4.op.3.running_mean", "cells.1._ops.4.op.3.running_var", "cells.1._ops.6.op.1.weight", "cells.1._ops.6.op.2.weight", "cells.1._ops.6.op.3.weight", "cells.1._ops.6.op.3.bias", "cells.1._ops.6.op.3.running_mean", "cells.1._ops.6.op.3.running_var", "cells.1._ops.6.op.5.weight", "cells.1._ops.6.op.6.weight", "cells.1._ops.6.op.7.weight", "cells.1._ops.6.op.7.bias", "cells.1._ops.6.op.7.running_mean", "cells.1._ops.6.op.7.running_var", "cells.1._ops.7.op.1.weight", "cells.1._ops.7.op.2.weight", "cells.1._ops.7.op.3.weight", "cells.1._ops.7.op.3.bias", "cells.1._ops.7.op.3.running_mean", "cells.1._ops.7.op.3.running_var", "cells.2.preprocess0.op.1.weight", "cells.2.preprocess0.op.2.weight", "cells.2.preprocess0.op.2.bias", "cells.2.preprocess0.op.2.running_mean", "cells.2.preprocess0.op.2.running_var", "cells.2.preprocess1.op.1.weight", "cells.2.preprocess1.op.2.weight", "cells.2.preprocess1.op.2.bias", "cells.2.preprocess1.op.2.running_mean", "cells.2.preprocess1.op.2.running_var", "cells.2._ops.0.op.1.weight", "cells.2._ops.0.op.2.weight", "cells.2._ops.0.op.3.weight", "cells.2._ops.0.op.3.bias", "cells.2._ops.0.op.3.running_mean", "cells.2._ops.0.op.3.running_var", "cells.2._ops.0.op.5.weight", "cells.2._ops.0.op.6.weight", "cells.2._ops.0.op.7.weight", "cells.2._ops.0.op.7.bias", "cells.2._ops.0.op.7.running_mean", "cells.2._ops.0.op.7.running_var", "cells.2._ops.1.op.1.weight", "cells.2._ops.1.op.2.weight", "cells.2._ops.1.op.3.weight", "cells.2._ops.1.op.3.bias", "cells.2._ops.1.op.3.running_mean", "cells.2._ops.1.op.3.running_var", "cells.2._ops.1.op.5.weight", "cells.2._ops.1.op.6.weight", "cells.2._ops.1.op.7.weight", "cells.2._ops.1.op.7.bias", "cells.2._ops.1.op.7.running_mean", "cells.2._ops.1.op.7.running_var", "cells.2._ops.2.op.1.weight", "cells.2._ops.2.op.2.weight", "cells.2._ops.2.op.3.weight", "cells.2._ops.2.op.3.bias", "cells.2._ops.2.op.3.running_mean", "cells.2._ops.2.op.3.running_var", "cells.2._ops.2.op.5.weight", "cells.2._ops.2.op.6.weight", "cells.2._ops.2.op.7.weight", "cells.2._ops.2.op.7.bias", "cells.2._ops.2.op.7.running_mean", "cells.2._ops.2.op.7.running_var", "cells.2._ops.3.op.1.weight", "cells.2._ops.3.op.2.weight", "cells.2._ops.3.op.3.weight", "cells.2._ops.3.op.3.bias", "cells.2._ops.3.op.3.running_mean", "cells.2._ops.3.op.3.running_var", "cells.2._ops.4.op.1.weight", "cells.2._ops.4.op.2.weight", "cells.2._ops.4.op.3.weight", "cells.2._ops.4.op.3.bias", "cells.2._ops.4.op.3.running_mean", "cells.2._ops.4.op.3.running_var", "cells.2._ops.6.op.1.weight", "cells.2._ops.6.op.2.weight", "cells.2._ops.6.op.3.weight", "cells.2._ops.6.op.3.bias", "cells.2._ops.6.op.3.running_mean", "cells.2._ops.6.op.3.running_var", "cells.2._ops.6.op.5.weight", "cells.2._ops.6.op.6.weight", "cells.2._ops.6.op.7.weight", "cells.2._ops.6.op.7.bias", "cells.2._ops.6.op.7.running_mean", "cells.2._ops.6.op.7.running_var", "cells.2._ops.7.op.1.weight", "cells.2._ops.7.op.2.weight", "cells.2._ops.7.op.3.weight", "cells.2._ops.7.op.3.bias", "cells.2._ops.7.op.3.running_mean", "cells.2._ops.7.op.3.running_var", "cells.3.preprocess0.op.1.weight", "cells.3.preprocess0.op.2.weight", "cells.3.preprocess0.op.2.bias", "cells.3.preprocess0.op.2.running_mean", "cells.3.preprocess0.op.2.running_var", "cells.3.preprocess1.op.1.weight", "cells.3.preprocess1.op.2.weight", "cells.3.preprocess1.op.2.bias", "cells.3.preprocess1.op.2.running_mean", "cells.3.preprocess1.op.2.running_var", "cells.3._ops.0.op.1.weight", "cells.3._ops.0.op.2.weight", "cells.3._ops.0.op.3.weight", "cells.3._ops.0.op.3.bias", "cells.3._ops.0.op.3.running_mean", "cells.3._ops.0.op.3.running_var", "cells.3._ops.0.op.5.weight", "cells.3._ops.0.op.6.weight", "cells.3._ops.0.op.7.weight", "cells.3._ops.0.op.7.bias", "cells.3._ops.0.op.7.running_mean", "cells.3._ops.0.op.7.running_var", "cells.3._ops.1.op.1.weight", "cells.3._ops.1.op.2.weight", "cells.3._ops.1.op.3.weight", "cells.3._ops.1.op.3.bias", "cells.3._ops.1.op.3.running_mean", "cells.3._ops.1.op.3.running_var", "cells.3._ops.1.op.5.weight", "cells.3._ops.1.op.6.weight", "cells.3._ops.1.op.7.weight", "cells.3._ops.1.op.7.bias", "cells.3._ops.1.op.7.running_mean", "cells.3._ops.1.op.7.running_var", "cells.3._ops.2.op.1.weight", "cells.3._ops.2.op.2.weight", "cells.3._ops.2.op.3.weight", "cells.3._ops.2.op.3.bias", "cells.3._ops.2.op.3.running_mean", "cells.3._ops.2.op.3.running_var", "cells.3._ops.2.op.5.weight", "cells.3._ops.2.op.6.weight", "cells.3._ops.2.op.7.weight", "cells.3._ops.2.op.7.bias", "cells.3._ops.2.op.7.running_mean", "cells.3._ops.2.op.7.running_var", "cells.3._ops.3.op.1.weight", "cells.3._ops.3.op.2.weight", "cells.3._ops.3.op.3.weight", "cells.3._ops.3.op.3.bias", "cells.3._ops.3.op.3.running_mean", "cells.3._ops.3.op.3.running_var", "cells.3._ops.4.op.1.weight", "cells.3._ops.4.op.2.weight", "cells.3._ops.4.op.3.weight", "cells.3._ops.4.op.3.bias", "cells.3._ops.4.op.3.running_mean", "cells.3._ops.4.op.3.running_var", "cells.3._ops.6.op.1.weight", "cells.3._ops.6.op.2.weight", "cells.3._ops.6.op.3.weight", "cells.3._ops.6.op.3.bias", "cells.3._ops.6.op.3.running_mean", "cells.3._ops.6.op.3.running_var", "cells.3._ops.6.op.5.weight", "cells.3._ops.6.op.6.weight", "cells.3._ops.6.op.7.weight", "cells.3._ops.6.op.7.bias", "cells.3._ops.6.op.7.running_mean", "cells.3._ops.6.op.7.running_var", "cells.3._ops.7.op.1.weight", "cells.3._ops.7.op.2.weight", "cells.3._ops.7.op.3.weight", "cells.3._ops.7.op.3.bias", "cells.3._ops.7.op.3.running_mean", "cells.3._ops.7.op.3.running_var", "cells.4.preprocess0.op.1.weight", "cells.4.preprocess0.op.2.weight", "cells.4.preprocess0.op.2.bias", "cells.4.preprocess0.op.2.running_mean", "cells.4.preprocess0.op.2.running_var", "cells.4.preprocess1.op.1.weight", "cells.4.preprocess1.op.2.weight", "cells.4.preprocess1.op.2.bias", "cells.4.preprocess1.op.2.running_mean", "cells.4.preprocess1.op.2.running_var", "cells.4._ops.0.op.1.weight", "cells.4._ops.0.op.2.weight", "cells.4._ops.0.op.3.weight", "cells.4._ops.0.op.3.bias", "cells.4._ops.0.op.3.running_mean", "cells.4._ops.0.op.3.running_var", "cells.4._ops.0.op.5.weight", "cells.4._ops.0.op.6.weight", "cells.4._ops.0.op.7.weight", "cells.4._ops.0.op.7.bias", "cells.4._ops.0.op.7.running_mean", "cells.4._ops.0.op.7.running_var", "cells.4._ops.1.op.1.weight", "cells.4._ops.1.op.2.weight", "cells.4._ops.1.op.3.weight", "cells.4._ops.1.op.3.bias", "cells.4._ops.1.op.3.running_mean", "cells.4._ops.1.op.3.running_var", "cells.4._ops.1.op.5.weight", "cells.4._ops.1.op.6.weight", "cells.4._ops.1.op.7.weight", "cells.4._ops.1.op.7.bias", "cells.4._ops.1.op.7.running_mean", "cells.4._ops.1.op.7.running_var", "cells.4._ops.2.op.1.weight", "cells.4._ops.2.op.2.weight", "cells.4._ops.2.op.3.weight", "cells.4._ops.2.op.3.bias", "cells.4._ops.2.op.3.running_mean", "cells.4._ops.2.op.3.running_var", "cells.4._ops.2.op.5.weight", "cells.4._ops.2.op.6.weight", "cells.4._ops.2.op.7.weight", "cells.4._ops.2.op.7.bias", "cells.4._ops.2.op.7.running_mean", "cells.4._ops.2.op.7.running_var", "cells.4._ops.3.op.1.weight", "cells.4._ops.3.op.2.weight", "cells.4._ops.3.op.3.weight", "cells.4._ops.3.op.3.bias", "cells.4._ops.3.op.3.running_mean", "cells.4._ops.3.op.3.running_var", "cells.4._ops.4.op.1.weight", "cells.4._ops.4.op.2.weight", "cells.4._ops.4.op.3.weight", "cells.4._ops.4.op.3.bias", "cells.4._ops.4.op.3.running_mean", "cells.4._ops.4.op.3.running_var", "cells.4._ops.6.op.1.weight", "cells.4._ops.6.op.2.weight", "cells.4._ops.6.op.3.weight", "cells.4._ops.6.op.3.bias", "cells.4._ops.6.op.3.running_mean", "cells.4._ops.6.op.3.running_var", "cells.4._ops.6.op.5.weight", "cells.4._ops.6.op.6.weight", "cells.4._ops.6.op.7.weight", "cells.4._ops.6.op.7.bias", "cells.4._ops.6.op.7.running_mean", "cells.4._ops.6.op.7.running_var", "cells.4._ops.7.op.1.weight", "cells.4._ops.7.op.2.weight", "cells.4._ops.7.op.3.weight", "cells.4._ops.7.op.3.bias", "cells.4._ops.7.op.3.running_mean", "cells.4._ops.7.op.3.running_var", "cells.5.preprocess0.op.1.weight", "cells.5.preprocess0.op.2.weight", "cells.5.preprocess0.op.2.bias", "cells.5.preprocess0.op.2.running_mean", "cells.5.preprocess0.op.2.running_var", "cells.5.preprocess1.op.1.weight", "cells.5.preprocess1.op.2.weight", "cells.5.preprocess1.op.2.bias", "cells.5.preprocess1.op.2.running_mean", "cells.5.preprocess1.op.2.running_var", "cells.5._ops.0.op.1.weight", "cells.5._ops.0.op.2.weight", "cells.5._ops.0.op.3.weight", "cells.5._ops.0.op.3.bias", "cells.5._ops.0.op.3.running_mean", "cells.5._ops.0.op.3.running_var", "cells.5._ops.0.op.5.weight", "cells.5._ops.0.op.6.weight", "cells.5._ops.0.op.7.weight", "cells.5._ops.0.op.7.bias", "cells.5._ops.0.op.7.running_mean", "cells.5._ops.0.op.7.running_var", "cells.5._ops.1.op.1.weight", "cells.5._ops.1.op.2.weight", "cells.5._ops.1.op.3.weight", "cells.5._ops.1.op.3.bias", "cells.5._ops.1.op.3.running_mean", "cells.5._ops.1.op.3.running_var", "cells.5._ops.1.op.5.weight", "cells.5._ops.1.op.6.weight", "cells.5._ops.1.op.7.weight", "cells.5._ops.1.op.7.bias", "cells.5._ops.1.op.7.running_mean", "cells.5._ops.1.op.7.running_var", "cells.5._ops.2.op.1.weight", "cells.5._ops.2.op.2.weight", "cells.5._ops.2.op.3.weight", "cells.5._ops.2.op.3.bias", "cells.5._ops.2.op.3.running_mean", "cells.5._ops.2.op.3.running_var", "cells.5._ops.2.op.5.weight", "cells.5._ops.2.op.6.weight", "cells.5._ops.2.op.7.weight", "cells.5._ops.2.op.7.bias", "cells.5._ops.2.op.7.running_mean", "cells.5._ops.2.op.7.running_var", "cells.5._ops.3.op.1.weight", "cells.5._ops.3.op.2.weight", "cells.5._ops.3.op.3.weight", "cells.5._ops.3.op.3.bias", "cells.5._ops.3.op.3.running_mean", "cells.5._ops.3.op.3.running_var", "cells.5._ops.4.op.1.weight", "cells.5._ops.4.op.2.weight", "cells.5._ops.4.op.3.weight", "cells.5._ops.4.op.3.bias", "cells.5._ops.4.op.3.running_mean", "cells.5._ops.4.op.3.running_var", "cells.5._ops.6.op.1.weight", "cells.5._ops.6.op.2.weight", "cells.5._ops.6.op.3.weight", "cells.5._ops.6.op.3.bias", "cells.5._ops.6.op.3.running_mean", "cells.5._ops.6.op.3.running_var", "cells.5._ops.6.op.5.weight", "cells.5._ops.6.op.6.weight", "cells.5._ops.6.op.7.weight", "cells.5._ops.6.op.7.bias", "cells.5._ops.6.op.7.running_mean", "cells.5._ops.6.op.7.running_var", "cells.5._ops.7.op.1.weight", "cells.5._ops.7.op.2.weight", "cells.5._ops.7.op.3.weight", "cells.5._ops.7.op.3.bias", "cells.5._ops.7.op.3.running_mean", "cells.5._ops.7.op.3.running_var", "cells.6.preprocess0.op.1.weight", "cells.6.preprocess0.op.2.weight", "cells.6.preprocess0.op.2.bias", "cells.6.preprocess0.op.2.running_mean", "cells.6.preprocess0.op.2.running_var", "cells.6.preprocess1.op.1.weight", "cells.6.preprocess1.op.2.weight", "cells.6.preprocess1.op.2.bias", "cells.6.preprocess1.op.2.running_mean", "cells.6.preprocess1.op.2.running_var", "cells.6._ops.0.op.1.weight", "cells.6._ops.0.op.2.weight", "cells.6._ops.0.op.3.weight", "cells.6._ops.0.op.3.bias", "cells.6._ops.0.op.3.running_mean", "cells.6._ops.0.op.3.running_var", "cells.6._ops.0.op.5.weight", "cells.6._ops.0.op.6.weight", "cells.6._ops.0.op.7.weight", "cells.6._ops.0.op.7.bias", "cells.6._ops.0.op.7.running_mean", "cells.6._ops.0.op.7.running_var", "cells.6._ops.2.op.1.weight", "cells.6._ops.2.op.2.weight", "cells.6._ops.2.op.3.weight", "cells.6._ops.2.op.3.bias", "cells.6._ops.2.op.3.running_mean", "cells.6._ops.2.op.3.running_var", "cells.6._ops.3.op.1.weight", "cells.6._ops.3.op.2.weight", "cells.6._ops.3.op.3.weight", "cells.6._ops.3.op.3.bias", "cells.6._ops.3.op.3.running_mean", "cells.6._ops.3.op.3.running_var", "cells.6._ops.3.op.5.weight", "cells.6._ops.3.op.6.weight", "cells.6._ops.3.op.7.weight", "cells.6._ops.3.op.7.bias", "cells.6._ops.3.op.7.running_mean", "cells.6._ops.3.op.7.running_var", "cells.6._ops.4.op.1.weight", "cells.6._ops.4.op.2.weight", "cells.6._ops.4.op.3.weight", "cells.6._ops.4.op.3.bias", "cells.6._ops.4.op.3.running_mean", "cells.6._ops.4.op.3.running_var", "cells.6._ops.4.op.5.weight", "cells.6._ops.4.op.6.weight", "cells.6._ops.4.op.7.weight", "cells.6._ops.4.op.7.bias", "cells.6._ops.4.op.7.running_mean", "cells.6._ops.4.op.7.running_var", "cells.6._ops.6.op.1.weight", "cells.6._ops.6.op.2.weight", "cells.6._ops.6.op.3.weight", "cells.6._ops.6.op.3.bias", "cells.6._ops.6.op.3.running_mean", "cells.6._ops.6.op.3.running_var", "cells.7.preprocess0.conv_1.weight", "cells.7.preprocess0.conv_2.weight", "cells.7.preprocess0.bn.weight", "cells.7.preprocess0.bn.bias", "cells.7.preprocess0.bn.running_mean", "cells.7.preprocess0.bn.running_var", "cells.7.preprocess1.op.1.weight", "cells.7.preprocess1.op.2.weight", "cells.7.preprocess1.op.2.bias", "cells.7.preprocess1.op.2.running_mean", "cells.7.preprocess1.op.2.running_var", "cells.7._ops.0.op.1.weight", "cells.7._ops.0.op.2.weight", "cells.7._ops.0.op.3.weight", "cells.7._ops.0.op.3.bias", "cells.7._ops.0.op.3.running_mean", "cells.7._ops.0.op.3.running_var", "cells.7._ops.0.op.5.weight", "cells.7._ops.0.op.6.weight", "cells.7._ops.0.op.7.weight", "cells.7._ops.0.op.7.bias", "cells.7._ops.0.op.7.running_mean", "cells.7._ops.0.op.7.running_var", "cells.7._ops.1.op.1.weight", "cells.7._ops.1.op.2.weight", "cells.7._ops.1.op.3.weight", "cells.7._ops.1.op.3.bias", "cells.7._ops.1.op.3.running_mean", "cells.7._ops.1.op.3.running_var", "cells.7._ops.1.op.5.weight", "cells.7._ops.1.op.6.weight", "cells.7._ops.1.op.7.weight", "cells.7._ops.1.op.7.bias", "cells.7._ops.1.op.7.running_mean", "cells.7._ops.1.op.7.running_var", "cells.7._ops.2.op.1.weight", "cells.7._ops.2.op.2.weight", "cells.7._ops.2.op.3.weight", "cells.7._ops.2.op.3.bias", "cells.7._ops.2.op.3.running_mean", "cells.7._ops.2.op.3.running_var", "cells.7._ops.2.op.5.weight", "cells.7._ops.2.op.6.weight", "cells.7._ops.2.op.7.weight", "cells.7._ops.2.op.7.bias", "cells.7._ops.2.op.7.running_mean", "cells.7._ops.2.op.7.running_var", "cells.7._ops.3.op.1.weight", "cells.7._ops.3.op.2.weight", "cells.7._ops.3.op.3.weight", "cells.7._ops.3.op.3.bias", "cells.7._ops.3.op.3.running_mean", "cells.7._ops.3.op.3.running_var", "cells.7._ops.4.op.1.weight", "cells.7._ops.4.op.2.weight", "cells.7._ops.4.op.3.weight", "cells.7._ops.4.op.3.bias", "cells.7._ops.4.op.3.running_mean", "cells.7._ops.4.op.3.running_var", "cells.7._ops.6.op.1.weight", "cells.7._ops.6.op.2.weight", "cells.7._ops.6.op.3.weight", "cells.7._ops.6.op.3.bias", "cells.7._ops.6.op.3.running_mean", "cells.7._ops.6.op.3.running_var", "cells.7._ops.6.op.5.weight", "cells.7._ops.6.op.6.weight", "cells.7._ops.6.op.7.weight", "cells.7._ops.6.op.7.bias", "cells.7._ops.6.op.7.running_mean", "cells.7._ops.6.op.7.running_var", "cells.7._ops.7.op.1.weight", "cells.7._ops.7.op.2.weight", "cells.7._ops.7.op.3.weight", "cells.7._ops.7.op.3.bias", "cells.7._ops.7.op.3.running_mean", "cells.7._ops.7.op.3.running_var", "cells.8.preprocess0.op.1.weight", "cells.8.preprocess0.op.2.weight", "cells.8.preprocess0.op.2.bias", "cells.8.preprocess0.op.2.running_mean", "cells.8.preprocess0.op.2.running_var", "cells.8.preprocess1.op.1.weight", "cells.8.preprocess1.op.2.weight", "cells.8.preprocess1.op.2.bias", "cells.8.preprocess1.op.2.running_mean", "cells.8.preprocess1.op.2.running_var", "cells.8._ops.0.op.1.weight", "cells.8._ops.0.op.2.weight", "cells.8._ops.0.op.3.weight", "cells.8._ops.0.op.3.bias", "cells.8._ops.0.op.3.running_mean", "cells.8._ops.0.op.3.running_var", "cells.8._ops.0.op.5.weight", "cells.8._ops.0.op.6.weight", "cells.8._ops.0.op.7.weight", "cells.8._ops.0.op.7.bias", "cells.8._ops.0.op.7.running_mean", "cells.8._ops.0.op.7.running_var", "cells.8._ops.1.op.1.weight", "cells.8._ops.1.op.2.weight", "cells.8._ops.1.op.3.weight", "cells.8._ops.1.op.3.bias", "cells.8._ops.1.op.3.running_mean", "cells.8._ops.1.op.3.running_var", "cells.8._ops.1.op.5.weight", "cells.8._ops.1.op.6.weight", "cells.8._ops.1.op.7.weight", "cells.8._ops.1.op.7.bias", "cells.8._ops.1.op.7.running_mean", "cells.8._ops.1.op.7.running_var", "cells.8._ops.2.op.1.weight", "cells.8._ops.2.op.2.weight", "cells.8._ops.2.op.3.weight", "cells.8._ops.2.op.3.bias", "cells.8._ops.2.op.3.running_mean", "cells.8._ops.2.op.3.running_var", "cells.8._ops.2.op.5.weight", "cells.8._ops.2.op.6.weight", "cells.8._ops.2.op.7.weight", "cells.8._ops.2.op.7.bias", "cells.8._ops.2.op.7.running_mean", "cells.8._ops.2.op.7.running_var", "cells.8._ops.3.op.1.weight", "cells.8._ops.3.op.2.weight", "cells.8._ops.3.op.3.weight", "cells.8._ops.3.op.3.bias", "cells.8._ops.3.op.3.running_mean", "cells.8._ops.3.op.3.running_var", "cells.8._ops.4.op.1.weight", "cells.8._ops.4.op.2.weight", "cells.8._ops.4.op.3.weight", "cells.8._ops.4.op.3.bias", "cells.8._ops.4.op.3.running_mean", "cells.8._ops.4.op.3.running_var", "cells.8._ops.6.op.1.weight", "cells.8._ops.6.op.2.weight", "cells.8._ops.6.op.3.weight", "cells.8._ops.6.op.3.bias", "cells.8._ops.6.op.3.running_mean", "cells.8._ops.6.op.3.running_var", "cells.8._ops.6.op.5.weight", "cells.8._ops.6.op.6.weight", "cells.8._ops.6.op.7.weight", "cells.8._ops.6.op.7.bias", "cells.8._ops.6.op.7.running_mean", "cells.8._ops.6.op.7.running_var", "cells.8._ops.7.op.1.weight", "cells.8._ops.7.op.2.weight", "cells.8._ops.7.op.3.weight", "cells.8._ops.7.op.3.bias", "cells.8._ops.7.op.3.running_mean", "cells.8._ops.7.op.3.running_var", "cells.9.preprocess0.op.1.weight", "cells.9.preprocess0.op.2.weight", "cells.9.preprocess0.op.2.bias", "cells.9.preprocess0.op.2.running_mean", "cells.9.preprocess0.op.2.running_var", "cells.9.preprocess1.op.1.weight", "cells.9.preprocess1.op.2.weight", "cells.9.preprocess1.op.2.bias", "cells.9.preprocess1.op.2.running_mean", "cells.9.preprocess1.op.2.running_var", "cells.9._ops.0.op.1.weight", "cells.9._ops.0.op.2.weight", "cells.9._ops.0.op.3.weight", "cells.9._ops.0.op.3.bias", "cells.9._ops.0.op.3.running_mean", "cells.9._ops.0.op.3.running_var", "cells.9._ops.0.op.5.weight", "cells.9._ops.0.op.6.weight", "cells.9._ops.0.op.7.weight", "cells.9._ops.0.op.7.bias", "cells.9._ops.0.op.7.running_mean", "cells.9._ops.0.op.7.running_var", "cells.9._ops.1.op.1.weight", "cells.9._ops.1.op.2.weight", "cells.9._ops.1.op.3.weight", "cells.9._ops.1.op.3.bias", "cells.9._ops.1.op.3.running_mean", "cells.9._ops.1.op.3.running_var", "cells.9._ops.1.op.5.weight", "cells.9._ops.1.op.6.weight", "cells.9._ops.1.op.7.weight", "cells.9._ops.1.op.7.bias", "cells.9._ops.1.op.7.running_mean", "cells.9._ops.1.op.7.running_var", "cells.9._ops.2.op.1.weight", "cells.9._ops.2.op.2.weight", "cells.9._ops.2.op.3.weight", "cells.9._ops.2.op.3.bias", "cells.9._ops.2.op.3.running_mean", "cells.9._ops.2.op.3.running_var", "cells.9._ops.2.op.5.weight", "cells.9._ops.2.op.6.weight", "cells.9._ops.2.op.7.weight", "cells.9._ops.2.op.7.bias", "cells.9._ops.2.op.7.running_mean", "cells.9._ops.2.op.7.running_var", "cells.9._ops.3.op.1.weight", "cells.9._ops.3.op.2.weight", "cells.9._ops.3.op.3.weight", "cells.9._ops.3.op.3.bias", "cells.9._ops.3.op.3.running_mean", "cells.9._ops.3.op.3.running_var", "cells.9._ops.4.op.1.weight", "cells.9._ops.4.op.2.weight", "cells.9._ops.4.op.3.weight", "cells.9._ops.4.op.3.bias", "cells.9._ops.4.op.3.running_mean", "cells.9._ops.4.op.3.running_var", "cells.9._ops.6.op.1.weight", "cells.9._ops.6.op.2.weight", "cells.9._ops.6.op.3.weight", "cells.9._ops.6.op.3.bias", "cells.9._ops.6.op.3.running_mean", "cells.9._ops.6.op.3.running_var", "cells.9._ops.6.op.5.weight", "cells.9._ops.6.op.6.weight", "cells.9._ops.6.op.7.weight", "cells.9._ops.6.op.7.bias", "cells.9._ops.6.op.7.running_mean", "cells.9._ops.6.op.7.running_var", "cells.9._ops.7.op.1.weight", "cells.9._ops.7.op.2.weight", "cells.9._ops.7.op.3.weight", "cells.9._ops.7.op.3.bias", "cells.9._ops.7.op.3.running_mean", "cells.9._ops.7.op.3.running_var", "cells.10.preprocess0.op.1.weight", "cells.10.preprocess0.op.2.weight", "cells.10.preprocess0.op.2.bias", "cells.10.preprocess0.op.2.running_mean", "cells.10.preprocess0.op.2.running_var", "cells.10.preprocess1.op.1.weight", "cells.10.preprocess1.op.2.weight", "cells.10.preprocess1.op.2.bias", "cells.10.preprocess1.op.2.running_mean", "cells.10.preprocess1.op.2.running_var", "cells.10._ops.0.op.1.weight", "cells.10._ops.0.op.2.weight", "cells.10._ops.0.op.3.weight", "cells.10._ops.0.op.3.bias", "cells.10._ops.0.op.3.running_mean", "cells.10._ops.0.op.3.running_var", "cells.10._ops.0.op.5.weight", "cells.10._ops.0.op.6.weight", "cells.10._ops.0.op.7.weight", "cells.10._ops.0.op.7.bias", "cells.10._ops.0.op.7.running_mean", "cells.10._ops.0.op.7.running_var", "cells.10._ops.1.op.1.weight", "cells.10._ops.1.op.2.weight", "cells.10._ops.1.op.3.weight", "cells.10._ops.1.op.3.bias", "cells.10._ops.1.op.3.running_mean", "cells.10._ops.1.op.3.running_var", "cells.10._ops.1.op.5.weight", "cells.10._ops.1.op.6.weight", "cells.10._ops.1.op.7.weight", "cells.10._ops.1.op.7.bias", "cells.10._ops.1.op.7.running_mean", "cells.10._ops.1.op.7.running_var", "cells.10._ops.2.op.1.weight", "cells.10._ops.2.op.2.weight", "cells.10._ops.2.op.3.weight", "cells.10._ops.2.op.3.bias", "cells.10._ops.2.op.3.running_mean", "cells.10._ops.2.op.3.running_var", "cells.10._ops.2.op.5.weight", "cells.10._ops.2.op.6.weight", "cells.10._ops.2.op.7.weight", "cells.10._ops.2.op.7.bias", "cells.10._ops.2.op.7.running_mean", "cells.10._ops.2.op.7.running_var", "cells.10._ops.3.op.1.weight", "cells.10._ops.3.op.2.weight", "cells.10._ops.3.op.3.weight", "cells.10._ops.3.op.3.bias", "cells.10._ops.3.op.3.running_mean", "cells.10._ops.3.op.3.running_var", "cells.10._ops.4.op.1.weight", "cells.10._ops.4.op.2.weight", "cells.10._ops.4.op.3.weight", "cells.10._ops.4.op.3.bias", "cells.10._ops.4.op.3.running_mean", "cells.10._ops.4.op.3.running_var", "cells.10._ops.6.op.1.weight", "cells.10._ops.6.op.2.weight", "cells.10._ops.6.op.3.weight", "cells.10._ops.6.op.3.bias", "cells.10._ops.6.op.3.running_mean", "cells.10._ops.6.op.3.running_var", "cells.10._ops.6.op.5.weight", "cells.10._ops.6.op.6.weight", "cells.10._ops.6.op.7.weight", "cells.10._ops.6.op.7.bias", "cells.10._ops.6.op.7.running_mean", "cells.10._ops.6.op.7.running_var", "cells.10._ops.7.op.1.weight", "cells.10._ops.7.op.2.weight", "cells.10._ops.7.op.3.weight", "cells.10._ops.7.op.3.bias", "cells.10._ops.7.op.3.running_mean", "cells.10._ops.7.op.3.running_var", "cells.11.preprocess0.op.1.weight", "cells.11.preprocess0.op.2.weight", "cells.11.preprocess0.op.2.bias", "cells.11.preprocess0.op.2.running_mean", "cells.11.preprocess0.op.2.running_var", "cells.11.preprocess1.op.1.weight", "cells.11.preprocess1.op.2.weight", "cells.11.preprocess1.op.2.bias", "cells.11.preprocess1.op.2.running_mean", "cells.11.preprocess1.op.2.running_var", "cells.11._ops.0.op.1.weight", "cells.11._ops.0.op.2.weight", "cells.11._ops.0.op.3.weight", "cells.11._ops.0.op.3.bias", "cells.11._ops.0.op.3.running_mean", "cells.11._ops.0.op.3.running_var", "cells.11._ops.0.op.5.weight", "cells.11._ops.0.op.6.weight", "cells.11._ops.0.op.7.weight", "cells.11._ops.0.op.7.bias", "cells.11._ops.0.op.7.running_mean", "cells.11._ops.0.op.7.running_var", "cells.11._ops.1.op.1.weight", "cells.11._ops.1.op.2.weight", "cells.11._ops.1.op.3.weight", "cells.11._ops.1.op.3.bias", "cells.11._ops.1.op.3.running_mean", "cells.11._ops.1.op.3.running_var", "cells.11._ops.1.op.5.weight", "cells.11._ops.1.op.6.weight", "cells.11._ops.1.op.7.weight", "cells.11._ops.1.op.7.bias", "cells.11._ops.1.op.7.running_mean", "cells.11._ops.1.op.7.running_var", "cells.11._ops.2.op.1.weight", "cells.11._ops.2.op.2.weight", "cells.11._ops.2.op.3.weight", "cells.11._ops.2.op.3.bias", "cells.11._ops.2.op.3.running_mean", "cells.11._ops.2.op.3.running_var", "cells.11._ops.2.op.5.weight", "cells.11._ops.2.op.6.weight", "cells.11._ops.2.op.7.weight", "cells.11._ops.2.op.7.bias", "cells.11._ops.2.op.7.running_mean", "cells.11._ops.2.op.7.running_var", "cells.11._ops.3.op.1.weight", "cells.11._ops.3.op.2.weight", "cells.11._ops.3.op.3.weight", "cells.11._ops.3.op.3.bias", "cells.11._ops.3.op.3.running_mean", "cells.11._ops.3.op.3.running_var", "cells.11._ops.4.op.1.weight", "cells.11._ops.4.op.2.weight", "cells.11._ops.4.op.3.weight", "cells.11._ops.4.op.3.bias", "cells.11._ops.4.op.3.running_mean", "cells.11._ops.4.op.3.running_var", "cells.11._ops.6.op.1.weight", "cells.11._ops.6.op.2.weight", "cells.11._ops.6.op.3.weight", "cells.11._ops.6.op.3.bias", "cells.11._ops.6.op.3.running_mean", "cells.11._ops.6.op.3.running_var", "cells.11._ops.6.op.5.weight", "cells.11._ops.6.op.6.weight", "cells.11._ops.6.op.7.weight", "cells.11._ops.6.op.7.bias", "cells.11._ops.6.op.7.running_mean", "cells.11._ops.6.op.7.running_var", "cells.11._ops.7.op.1.weight", "cells.11._ops.7.op.2.weight", "cells.11._ops.7.op.3.weight", "cells.11._ops.7.op.3.bias", "cells.11._ops.7.op.3.running_mean", "cells.11._ops.7.op.3.running_var", "cells.12.preprocess0.op.1.weight", "cells.12.preprocess0.op.2.weight", "cells.12.preprocess0.op.2.bias", "cells.12.preprocess0.op.2.running_mean", "cells.12.preprocess0.op.2.running_var", "cells.12.preprocess1.op.1.weight", "cells.12.preprocess1.op.2.weight", "cells.12.preprocess1.op.2.bias", "cells.12.preprocess1.op.2.running_mean", "cells.12.preprocess1.op.2.running_var", "cells.12._ops.0.op.1.weight", "cells.12._ops.0.op.2.weight", "cells.12._ops.0.op.3.weight", "cells.12._ops.0.op.3.bias", "cells.12._ops.0.op.3.running_mean", "cells.12._ops.0.op.3.running_var", "cells.12._ops.0.op.5.weight", "cells.12._ops.0.op.6.weight", "cells.12._ops.0.op.7.weight", "cells.12._ops.0.op.7.bias", "cells.12._ops.0.op.7.running_mean", "cells.12._ops.0.op.7.running_var", "cells.12._ops.1.op.1.weight", "cells.12._ops.1.op.2.weight", "cells.12._ops.1.op.3.weight", "cells.12._ops.1.op.3.bias", "cells.12._ops.1.op.3.running_mean", "cells.12._ops.1.op.3.running_var", "cells.12._ops.1.op.5.weight", "cells.12._ops.1.op.6.weight", "cells.12._ops.1.op.7.weight", "cells.12._ops.1.op.7.bias", "cells.12._ops.1.op.7.running_mean", "cells.12._ops.1.op.7.running_var", "cells.12._ops.2.op.1.weight", "cells.12._ops.2.op.2.weight", "cells.12._ops.2.op.3.weight", "cells.12._ops.2.op.3.bias", "cells.12._ops.2.op.3.running_mean", "cells.12._ops.2.op.3.running_var", "cells.12._ops.2.op.5.weight", "cells.12._ops.2.op.6.weight", "cells.12._ops.2.op.7.weight", "cells.12._ops.2.op.7.bias", "cells.12._ops.2.op.7.running_mean", "cells.12._ops.2.op.7.running_var", "cells.12._ops.3.op.1.weight", "cells.12._ops.3.op.2.weight", "cells.12._ops.3.op.3.weight", "cells.12._ops.3.op.3.bias", "cells.12._ops.3.op.3.running_mean", "cells.12._ops.3.op.3.running_var", "cells.12._ops.4.op.1.weight", "cells.12._ops.4.op.2.weight", "cells.12._ops.4.op.3.weight", "cells.12._ops.4.op.3.bias", "cells.12._ops.4.op.3.running_mean", "cells.12._ops.4.op.3.running_var", "cells.12._ops.6.op.1.weight", "cells.12._ops.6.op.2.weight", "cells.12._ops.6.op.3.weight", "cells.12._ops.6.op.3.bias", "cells.12._ops.6.op.3.running_mean", "cells.12._ops.6.op.3.running_var", "cells.12._ops.6.op.5.weight", "cells.12._ops.6.op.6.weight", "cells.12._ops.6.op.7.weight", "cells.12._ops.6.op.7.bias", "cells.12._ops.6.op.7.running_mean", "cells.12._ops.6.op.7.running_var", "cells.12._ops.7.op.1.weight", "cells.12._ops.7.op.2.weight", "cells.12._ops.7.op.3.weight", "cells.12._ops.7.op.3.bias", "cells.12._ops.7.op.3.running_mean", "cells.12._ops.7.op.3.running_var", "cells.13.preprocess0.op.1.weight", "cells.13.preprocess0.op.2.weight", "cells.13.preprocess0.op.2.bias", "cells.13.preprocess0.op.2.running_mean", "cells.13.preprocess0.op.2.running_var", "cells.13.preprocess1.op.1.weight", "cells.13.preprocess1.op.2.weight", "cells.13.preprocess1.op.2.bias", "cells.13.preprocess1.op.2.running_mean", "cells.13.preprocess1.op.2.running_var", "cells.13._ops.0.op.1.weight", "cells.13._ops.0.op.2.weight", "cells.13._ops.0.op.3.weight", "cells.13._ops.0.op.3.bias", "cells.13._ops.0.op.3.running_mean", "cells.13._ops.0.op.3.running_var", "cells.13._ops.0.op.5.weight", "cells.13._ops.0.op.6.weight", "cells.13._ops.0.op.7.weight", "cells.13._ops.0.op.7.bias", "cells.13._ops.0.op.7.running_mean", "cells.13._ops.0.op.7.running_var", "cells.13._ops.2.op.1.weight", "cells.13._ops.2.op.2.weight", "cells.13._ops.2.op.3.weight", "cells.13._ops.2.op.3.bias", "cells.13._ops.2.op.3.running_mean", "cells.13._ops.2.op.3.running_var", "cells.13._ops.3.op.1.weight", "cells.13._ops.3.op.2.weight", "cells.13._ops.3.op.3.weight", "cells.13._ops.3.op.3.bias", "cells.13._ops.3.op.3.running_mean", "cells.13._ops.3.op.3.running_var", "cells.13._ops.3.op.5.weight", "cells.13._ops.3.op.6.weight", "cells.13._ops.3.op.7.weight", "cells.13._ops.3.op.7.bias", "cells.13._ops.3.op.7.running_mean", "cells.13._ops.3.op.7.running_var", "cells.13._ops.4.op.1.weight", "cells.13._ops.4.op.2.weight", "cells.13._ops.4.op.3.weight", "cells.13._ops.4.op.3.bias", "cells.13._ops.4.op.3.running_mean", "cells.13._ops.4.op.3.running_var", "cells.13._ops.4.op.5.weight", "cells.13._ops.4.op.6.weight", "cells.13._ops.4.op.7.weight", "cells.13._ops.4.op.7.bias", "cells.13._ops.4.op.7.running_mean", "cells.13._ops.4.op.7.running_var", "cells.13._ops.6.op.1.weight", "cells.13._ops.6.op.2.weight", "cells.13._ops.6.op.3.weight", "cells.13._ops.6.op.3.bias", "cells.13._ops.6.op.3.running_mean", "cells.13._ops.6.op.3.running_var", "cells.14.preprocess0.conv_1.weight", "cells.14.preprocess0.conv_2.weight", "cells.14.preprocess0.bn.weight", "cells.14.preprocess0.bn.bias", "cells.14.preprocess0.bn.running_mean", "cells.14.preprocess0.bn.running_var", "cells.14.preprocess1.op.1.weight", "cells.14.preprocess1.op.2.weight", "cells.14.preprocess1.op.2.bias", "cells.14.preprocess1.op.2.running_mean", "cells.14.preprocess1.op.2.running_var", "cells.14._ops.0.op.1.weight", "cells.14._ops.0.op.2.weight", "cells.14._ops.0.op.3.weight", "cells.14._ops.0.op.3.bias", "cells.14._ops.0.op.3.running_mean", "cells.14._ops.0.op.3.running_var", "cells.14._ops.0.op.5.weight", "cells.14._ops.0.op.6.weight", "cells.14._ops.0.op.7.weight", "cells.14._ops.0.op.7.bias", "cells.14._ops.0.op.7.running_mean", "cells.14._ops.0.op.7.running_var", "cells.14._ops.1.op.1.weight", "cells.14._ops.1.op.2.weight", "cells.14._ops.1.op.3.weight", "cells.14._ops.1.op.3.bias", "cells.14._ops.1.op.3.running_mean", "cells.14._ops.1.op.3.running_var", "cells.14._ops.1.op.5.weight", "cells.14._ops.1.op.6.weight", "cells.14._ops.1.op.7.weight", "cells.14._ops.1.op.7.bias", "cells.14._ops.1.op.7.running_mean", "cells.14._ops.1.op.7.running_var", "cells.14._ops.2.op.1.weight", "cells.14._ops.2.op.2.weight", "cells.14._ops.2.op.3.weight", "cells.14._ops.2.op.3.bias", "cells.14._ops.2.op.3.running_mean", "cells.14._ops.2.op.3.running_var", "cells.14._ops.2.op.5.weight", "cells.14._ops.2.op.6.weight", "cells.14._ops.2.op.7.weight", "cells.14._ops.2.op.7.bias", "cells.14._ops.2.op.7.running_mean", "cells.14._ops.2.op.7.running_var", "cells.14._ops.3.op.1.weight", "cells.14._ops.3.op.2.weight", "cells.14._ops.3.op.3.weight", "cells.14._ops.3.op.3.bias", "cells.14._ops.3.op.3.running_mean", "cells.14._ops.3.op.3.running_var", "cells.14._ops.4.op.1.weight", "cells.14._ops.4.op.2.weight", "cells.14._ops.4.op.3.weight", "cells.14._ops.4.op.3.bias", "cells.14._ops.4.op.3.running_mean", "cells.14._ops.4.op.3.running_var", "cells.14._ops.6.op.1.weight", "cells.14._ops.6.op.2.weight", "cells.14._ops.6.op.3.weight", "cells.14._ops.6.op.3.bias", "cells.14._ops.6.op.3.running_mean", "cells.14._ops.6.op.3.running_var", "cells.14._ops.6.op.5.weight", "cells.14._ops.6.op.6.weight", "cells.14._ops.6.op.7.weight", "cells.14._ops.6.op.7.bias", "cells.14._ops.6.op.7.running_mean", "cells.14._ops.6.op.7.running_var", "cells.14._ops.7.op.1.weight", "cells.14._ops.7.op.2.weight", "cells.14._ops.7.op.3.weight", "cells.14._ops.7.op.3.bias", "cells.14._ops.7.op.3.running_mean", "cells.14._ops.7.op.3.running_var", "cells.15.preprocess0.op.1.weight", "cells.15.preprocess0.op.2.weight", "cells.15.preprocess0.op.2.bias", "cells.15.preprocess0.op.2.running_mean", "cells.15.preprocess0.op.2.running_var", "cells.15.preprocess1.op.1.weight", "cells.15.preprocess1.op.2.weight", "cells.15.preprocess1.op.2.bias", "cells.15.preprocess1.op.2.running_mean", "cells.15.preprocess1.op.2.running_var", "cells.15._ops.0.op.1.weight", "cells.15._ops.0.op.2.weight", "cells.15._ops.0.op.3.weight", "cells.15._ops.0.op.3.bias", "cells.15._ops.0.op.3.running_mean", "cells.15._ops.0.op.3.running_var", "cells.15._ops.0.op.5.weight", "cells.15._ops.0.op.6.weight", "cells.15._ops.0.op.7.weight", "cells.15._ops.0.op.7.bias", "cells.15._ops.0.op.7.running_mean", "cells.15._ops.0.op.7.running_var", "cells.15._ops.1.op.1.weight", "cells.15._ops.1.op.2.weight", "cells.15._ops.1.op.3.weight", "cells.15._ops.1.op.3.bias", "cells.15._ops.1.op.3.running_mean", "cells.15._ops.1.op.3.running_var", "cells.15._ops.1.op.5.weight", "cells.15._ops.1.op.6.weight", "cells.15._ops.1.op.7.weight", "cells.15._ops.1.op.7.bias", "cells.15._ops.1.op.7.running_mean", "cells.15._ops.1.op.7.running_var", "cells.15._ops.2.op.1.weight", "cells.15._ops.2.op.2.weight", "cells.15._ops.2.op.3.weight", "cells.15._ops.2.op.3.bias", "cells.15._ops.2.op.3.running_mean", "cells.15._ops.2.op.3.running_var", "cells.15._ops.2.op.5.weight", "cells.15._ops.2.op.6.weight", "cells.15._ops.2.op.7.weight", "cells.15._ops.2.op.7.bias", "cells.15._ops.2.op.7.running_mean", "cells.15._ops.2.op.7.running_var", "cells.15._ops.3.op.1.weight", "cells.15._ops.3.op.2.weight", "cells.15._ops.3.op.3.weight", "cells.15._ops.3.op.3.bias", "cells.15._ops.3.op.3.running_mean", "cells.15._ops.3.op.3.running_var", "cells.15._ops.4.op.1.weight", "cells.15._ops.4.op.2.weight", "cells.15._ops.4.op.3.weight", "cells.15._ops.4.op.3.bias", "cells.15._ops.4.op.3.running_mean", "cells.15._ops.4.op.3.running_var", "cells.15._ops.6.op.1.weight", "cells.15._ops.6.op.2.weight", "cells.15._ops.6.op.3.weight", "cells.15._ops.6.op.3.bias", "cells.15._ops.6.op.3.running_mean", "cells.15._ops.6.op.3.running_var", "cells.15._ops.6.op.5.weight", "cells.15._ops.6.op.6.weight", "cells.15._ops.6.op.7.weight", "cells.15._ops.6.op.7.bias", "cells.15._ops.6.op.7.running_mean", "cells.15._ops.6.op.7.running_var", "cells.15._ops.7.op.1.weight", "cells.15._ops.7.op.2.weight", "cells.15._ops.7.op.3.weight", "cells.15._ops.7.op.3.bias", "cells.15._ops.7.op.3.running_mean", "cells.15._ops.7.op.3.running_var", "cells.16.preprocess0.op.1.weight", "cells.16.preprocess0.op.2.weight", "cells.16.preprocess0.op.2.bias", "cells.16.preprocess0.op.2.running_mean", "cells.16.preprocess0.op.2.running_var", "cells.16.preprocess1.op.1.weight", "cells.16.preprocess1.op.2.weight", "cells.16.preprocess1.op.2.bias", "cells.16.preprocess1.op.2.running_mean", "cells.16.preprocess1.op.2.running_var", "cells.16._ops.0.op.1.weight", "cells.16._ops.0.op.2.weight", "cells.16._ops.0.op.3.weight", "cells.16._ops.0.op.3.bias", "cells.16._ops.0.op.3.running_mean", "cells.16._ops.0.op.3.running_var", "cells.16._ops.0.op.5.weight", "cells.16._ops.0.op.6.weight", "cells.16._ops.0.op.7.weight", "cells.16._ops.0.op.7.bias", "cells.16._ops.0.op.7.running_mean", "cells.16._ops.0.op.7.running_var", "cells.16._ops.1.op.1.weight", "cells.16._ops.1.op.2.weight", "cells.16._ops.1.op.3.weight", "cells.16._ops.1.op.3.bias", "cells.16._ops.1.op.3.running_mean", "cells.16._ops.1.op.3.running_var", "cells.16._ops.1.op.5.weight", "cells.16._ops.1.op.6.weight", "cells.16._ops.1.op.7.weight", "cells.16._ops.1.op.7.bias", "cells.16._ops.1.op.7.running_mean", "cells.16._ops.1.op.7.running_var", "cells.16._ops.2.op.1.weight", "cells.16._ops.2.op.2.weight", "cells.16._ops.2.op.3.weight", "cells.16._ops.2.op.3.bias", "cells.16._ops.2.op.3.running_mean", "cells.16._ops.2.op.3.running_var", "cells.16._ops.2.op.5.weight", "cells.16._ops.2.op.6.weight", "cells.16._ops.2.op.7.weight", "cells.16._ops.2.op.7.bias", "cells.16._ops.2.op.7.running_mean", "cells.16._ops.2.op.7.running_var", "cells.16._ops.3.op.1.weight", "cells.16._ops.3.op.2.weight", "cells.16._ops.3.op.3.weight", "cells.16._ops.3.op.3.bias", "cells.16._ops.3.op.3.running_mean", "cells.16._ops.3.op.3.running_var", "cells.16._ops.4.op.1.weight", "cells.16._ops.4.op.2.weight", "cells.16._ops.4.op.3.weight", "cells.16._ops.4.op.3.bias", "cells.16._ops.4.op.3.running_mean", "cells.16._ops.4.op.3.running_var", "cells.16._ops.6.op.1.weight", "cells.16._ops.6.op.2.weight", "cells.16._ops.6.op.3.weight", "cells.16._ops.6.op.3.bias", "cells.16._ops.6.op.3.running_mean", "cells.16._ops.6.op.3.running_var", "cells.16._ops.6.op.5.weight", "cells.16._ops.6.op.6.weight", "cells.16._ops.6.op.7.weight", "cells.16._ops.6.op.7.bias", "cells.16._ops.6.op.7.running_mean", "cells.16._ops.6.op.7.running_var", "cells.16._ops.7.op.1.weight", "cells.16._ops.7.op.2.weight", "cells.16._ops.7.op.3.weight", "cells.16._ops.7.op.3.bias", "cells.16._ops.7.op.3.running_mean", "cells.16._ops.7.op.3.running_var", "cells.17.preprocess0.op.1.weight", "cells.17.preprocess0.op.2.weight", "cells.17.preprocess0.op.2.bias", "cells.17.preprocess0.op.2.running_mean", "cells.17.preprocess0.op.2.running_var", "cells.17.preprocess1.op.1.weight", "cells.17.preprocess1.op.2.weight", "cells.17.preprocess1.op.2.bias", "cells.17.preprocess1.op.2.running_mean", "cells.17.preprocess1.op.2.running_var", "cells.17._ops.0.op.1.weight", "cells.17._ops.0.op.2.weight", "cells.17._ops.0.op.3.weight", "cells.17._ops.0.op.3.bias", "cells.17._ops.0.op.3.running_mean", "cells.17._ops.0.op.3.running_var", "cells.17._ops.0.op.5.weight", "cells.17._ops.0.op.6.weight", "cells.17._ops.0.op.7.weight", "cells.17._ops.0.op.7.bias", "cells.17._ops.0.op.7.running_mean", "cells.17._ops.0.op.7.running_var", "cells.17._ops.1.op.1.weight", "cells.17._ops.1.op.2.weight", "cells.17._ops.1.op.3.weight", "cells.17._ops.1.op.3.bias", "cells.17._ops.1.op.3.running_mean", "cells.17._ops.1.op.3.running_var", "cells.17._ops.1.op.5.weight", "cells.17._ops.1.op.6.weight", "cells.17._ops.1.op.7.weight", "cells.17._ops.1.op.7.bias", "cells.17._ops.1.op.7.running_mean", "cells.17._ops.1.op.7.running_var", "cells.17._ops.2.op.1.weight", "cells.17._ops.2.op.2.weight", "cells.17._ops.2.op.3.weight", "cells.17._ops.2.op.3.bias", "cells.17._ops.2.op.3.running_mean", "cells.17._ops.2.op.3.running_var", "cells.17._ops.2.op.5.weight", "cells.17._ops.2.op.6.weight", "cells.17._ops.2.op.7.weight", "cells.17._ops.2.op.7.bias", "cells.17._ops.2.op.7.running_mean", "cells.17._ops.2.op.7.running_var", "cells.17._ops.3.op.1.weight", "cells.17._ops.3.op.2.weight", "cells.17._ops.3.op.3.weight", "cells.17._ops.3.op.3.bias", "cells.17._ops.3.op.3.running_mean", "cells.17._ops.3.op.3.running_var", "cells.17._ops.4.op.1.weight", "cells.17._ops.4.op.2.weight", "cells.17._ops.4.op.3.weight", "cells.17._ops.4.op.3.bias", "cells.17._ops.4.op.3.running_mean", "cells.17._ops.4.op.3.running_var", "cells.17._ops.6.op.1.weight", "cells.17._ops.6.op.2.weight", "cells.17._ops.6.op.3.weight", "cells.17._ops.6.op.3.bias", "cells.17._ops.6.op.3.running_mean", "cells.17._ops.6.op.3.running_var", "cells.17._ops.6.op.5.weight", "cells.17._ops.6.op.6.weight", "cells.17._ops.6.op.7.weight", "cells.17._ops.6.op.7.bias", "cells.17._ops.6.op.7.running_mean", "cells.17._ops.6.op.7.running_var", "cells.17._ops.7.op.1.weight", "cells.17._ops.7.op.2.weight", "cells.17._ops.7.op.3.weight", "cells.17._ops.7.op.3.bias", "cells.17._ops.7.op.3.running_mean", "cells.17._ops.7.op.3.running_var", "cells.18.preprocess0.op.1.weight", "cells.18.preprocess0.op.2.weight", "cells.18.preprocess0.op.2.bias", "cells.18.preprocess0.op.2.running_mean", "cells.18.preprocess0.op.2.running_var", "cells.18.preprocess1.op.1.weight", "cells.18.preprocess1.op.2.weight", "cells.18.preprocess1.op.2.bias", "cells.18.preprocess1.op.2.running_mean", "cells.18.preprocess1.op.2.running_var", "cells.18._ops.0.op.1.weight", "cells.18._ops.0.op.2.weight", "cells.18._ops.0.op.3.weight", "cells.18._ops.0.op.3.bias", "cells.18._ops.0.op.3.running_mean", "cells.18._ops.0.op.3.running_var", "cells.18._ops.0.op.5.weight", "cells.18._ops.0.op.6.weight", "cells.18._ops.0.op.7.weight", "cells.18._ops.0.op.7.bias", "cells.18._ops.0.op.7.running_mean", "cells.18._ops.0.op.7.running_var", "cells.18._ops.1.op.1.weight", "cells.18._ops.1.op.2.weight", "cells.18._ops.1.op.3.weight", "cells.18._ops.1.op.3.bias", "cells.18._ops.1.op.3.running_mean", "cells.18._ops.1.op.3.running_var", "cells.18._ops.1.op.5.weight", "cells.18._ops.1.op.6.weight", "cells.18._ops.1.op.7.weight", "cells.18._ops.1.op.7.bias", "cells.18._ops.1.op.7.running_mean", "cells.18._ops.1.op.7.running_var", "cells.18._ops.2.op.1.weight", "cells.18._ops.2.op.2.weight", "cells.18._ops.2.op.3.weight", "cells.18._ops.2.op.3.bias", "cells.18._ops.2.op.3.running_mean", "cells.18._ops.2.op.3.running_var", "cells.18._ops.2.op.5.weight", "cells.18._ops.2.op.6.weight", "cells.18._ops.2.op.7.weight", "cells.18._ops.2.op.7.bias", "cells.18._ops.2.op.7.running_mean", "cells.18._ops.2.op.7.running_var", "cells.18._ops.3.op.1.weight", "cells.18._ops.3.op.2.weight", "cells.18._ops.3.op.3.weight", "cells.18._ops.3.op.3.bias", "cells.18._ops.3.op.3.running_mean", "cells.18._ops.3.op.3.running_var", "cells.18._ops.4.op.1.weight", "cells.18._ops.4.op.2.weight", "cells.18._ops.4.op.3.weight", "cells.18._ops.4.op.3.bias", "cells.18._ops.4.op.3.running_mean", "cells.18._ops.4.op.3.running_var", "cells.18._ops.6.op.1.weight", "cells.18._ops.6.op.2.weight", "cells.18._ops.6.op.3.weight", "cells.18._ops.6.op.3.bias", "cells.18._ops.6.op.3.running_mean", "cells.18._ops.6.op.3.running_var", "cells.18._ops.6.op.5.weight", "cells.18._ops.6.op.6.weight", "cells.18._ops.6.op.7.weight", "cells.18._ops.6.op.7.bias", "cells.18._ops.6.op.7.running_mean", "cells.18._ops.6.op.7.running_var", "cells.18._ops.7.op.1.weight", "cells.18._ops.7.op.2.weight", "cells.18._ops.7.op.3.weight", "cells.18._ops.7.op.3.bias", "cells.18._ops.7.op.3.running_mean", "cells.18._ops.7.op.3.running_var", "cells.19.preprocess0.op.1.weight", "cells.19.preprocess0.op.2.weight", "cells.19.preprocess0.op.2.bias", "cells.19.preprocess0.op.2.running_mean", "cells.19.preprocess0.op.2.running_var", "cells.19.preprocess1.op.1.weight", "cells.19.preprocess1.op.2.weight", "cells.19.preprocess1.op.2.bias", "cells.19.preprocess1.op.2.running_mean", "cells.19.preprocess1.op.2.running_var", "cells.19._ops.0.op.1.weight", "cells.19._ops.0.op.2.weight", "cells.19._ops.0.op.3.weight", "cells.19._ops.0.op.3.bias", "cells.19._ops.0.op.3.running_mean", "cells.19._ops.0.op.3.running_var", "cells.19._ops.0.op.5.weight", "cells.19._ops.0.op.6.weight", "cells.19._ops.0.op.7.weight", "cells.19._ops.0.op.7.bias", "cells.19._ops.0.op.7.running_mean", "cells.19._ops.0.op.7.running_var", "cells.19._ops.1.op.1.weight", "cells.19._ops.1.op.2.weight", "cells.19._ops.1.op.3.weight", "cells.19._ops.1.op.3.bias", "cells.19._ops.1.op.3.running_mean", "cells.19._ops.1.op.3.running_var", "cells.19._ops.1.op.5.weight", "cells.19._ops.1.op.6.weight", "cells.19._ops.1.op.7.weight", "cells.19._ops.1.op.7.bias", "cells.19._ops.1.op.7.running_mean", "cells.19._ops.1.op.7.running_var", "cells.19._ops.2.op.1.weight", "cells.19._ops.2.op.2.weight", "cells.19._ops.2.op.3.weight", "cells.19._ops.2.op.3.bias", "cells.19._ops.2.op.3.running_mean", "cells.19._ops.2.op.3.running_var", "cells.19._ops.2.op.5.weight", "cells.19._ops.2.op.6.weight", "cells.19._ops.2.op.7.weight", "cells.19._ops.2.op.7.bias", "cells.19._ops.2.op.7.running_mean", "cells.19._ops.2.op.7.running_var", "cells.19._ops.3.op.1.weight", "cells.19._ops.3.op.2.weight", "cells.19._ops.3.op.3.weight", "cells.19._ops.3.op.3.bias", "cells.19._ops.3.op.3.running_mean", "cells.19._ops.3.op.3.running_var", "cells.19._ops.4.op.1.weight", "cells.19._ops.4.op.2.weight", "cells.19._ops.4.op.3.weight", "cells.19._ops.4.op.3.bias", "cells.19._ops.4.op.3.running_mean", "cells.19._ops.4.op.3.running_var", "cells.19._ops.6.op.1.weight", "cells.19._ops.6.op.2.weight", "cells.19._ops.6.op.3.weight", "cells.19._ops.6.op.3.bias", "cells.19._ops.6.op.3.running_mean", "cells.19._ops.6.op.3.running_var", "cells.19._ops.6.op.5.weight", "cells.19._ops.6.op.6.weight", "cells.19._ops.6.op.7.weight", "cells.19._ops.6.op.7.bias", "cells.19._ops.6.op.7.running_mean", "cells.19._ops.6.op.7.running_var", "cells.19._ops.7.op.1.weight", "cells.19._ops.7.op.2.weight", "cells.19._ops.7.op.3.weight", "cells.19._ops.7.op.3.bias", "cells.19._ops.7.op.3.running_mean", "cells.19._ops.7.op.3.running_var", "auxiliary_head.features.2.weight", "auxiliary_head.features.3.weight", "auxiliary_head.features.3.bias", "auxiliary_head.features.3.running_mean", "auxiliary_head.features.3.running_var", "auxiliary_head.features.5.weight", "auxiliary_head.features.6.weight", "auxiliary_head.features.6.bias", "auxiliary_head.features.6.running_mean", "auxiliary_head.features.6.running_var", "auxiliary_head.classifier.weight", "auxiliary_head.classifier.bias", "classifier.weight", "classifier.bias". Unexpected key(s) in state_dict: "epoch", "state_dict", "best_acc", "optimizer".
test.py 84 lines
hjyoo@neuro:~/PycharmProjects/Shapley-NAS$ python test.py --model_path 'eval-train_cifar10-20221115-131311/res18_best.pth.tar' --arch 'shapley_nas' --data '/home/hjyoo/Datasets/cifar-10-batches-py/' --cutout --auxiliary File "/home/hjyoo/PycharmProjects/Shapley-NAS/test.py", line 84 target = Variable(target, volatile=True).cuda(async=True) ^ SyntaxError: invalid syntax
test.py 91~93 lines
hjyoo@neuro:~/PycharmProjects/Shapley-NAS$ python test.py --model_path 'eval-train_cifar10-20221115-131311/res18_best.pth.tar' --arch 'shapley_nas' --data '/home/hjyoo/Datasets/cifar-10-batches-py/' --cutout --auxiliary 11/16 03:31:16 PM gpu device = 0 11/16 03:31:16 PM args = Namespace(data='/home/hjyoo/Datasets/cifar-10-batches-py/', batch_size=96, report_freq=50, gpu=0, init_channels=36, layers=20, model_path='eval-train_cifar10-20221115-131311/res18_best.pth.tar', auxiliary=True, cutout=True, cutout_length=16, drop_path_prob=0.2, seed=0, arch='shapley_nas') 108 108 36 108 144 36 144 144 36 144 144 36 144 144 36 144 144 36 144 144 72 144 288 72 288 288 72 288 288 72 288 288 72 288 288 72 288 288 72 288 288 144 288 576 144 576 576 144 576 576 144 576 576 144 576 576 144 576 576 144 11/16 03:31:25 PM param size = 3.603358MB Files already downloaded and verified /home/hjyoo/PycharmProjects/Shapley-NAS/test.py:83: UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead. input = Variable(input, volatile=True).cuda() /home/hjyoo/PycharmProjects/Shapley-NAS/test.py:84: UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead. target = Variable(target, volatile=True).cuda(non_blocking=True) Traceback (most recent call last): File "/home/hjyoo/PycharmProjects/Shapley-NAS/test.py", line 102, in <module> main() File "/home/hjyoo/PycharmProjects/Shapley-NAS/test.py", line 72, in main test_acc, test_obj = infer(test_queue, model, criterion) File "/home/hjyoo/PycharmProjects/Shapley-NAS/test.py", line 91, in infer objs.update(loss.data[0], n) IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to convert a 0-dim tensor to a number
hjyoo@neuro:~/PycharmProjects/Shapley-NAS$ python test.py --model_path 'eval-train_cifar10-20221115-131311/res18_best.pth.tar' --arch 'shapley_nas' --data '/home/hjyoo/Datasets/cifar-10-batches-py/' --cutout --auxiliary 11/16 03:32:23 PM gpu device = 0 11/16 03:32:23 PM args = Namespace(data='/home/hjyoo/Datasets/cifar-10-batches-py/', batch_size=96, report_freq=50, gpu=0, init_channels=36, layers=20, model_path='eval-train_cifar10-20221115-131311/res18_best.pth.tar', auxiliary=True, cutout=True, cutout_length=16, drop_path_prob=0.2, seed=0, arch='shapley_nas') 108 108 36 108 144 36 144 144 36 144 144 36 144 144 36 144 144 36 144 144 72 144 288 72 288 288 72 288 288 72 288 288 72 288 288 72 288 288 72 288 288 144 288 576 144 576 576 144 576 576 144 576 576 144 576 576 144 576 576 144 11/16 03:32:31 PM param size = 3.603358MB Files already downloaded and verified /home/hjyoo/PycharmProjects/Shapley-NAS/test.py:83: UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead. input = Variable(input, volatile=True).cuda() /home/hjyoo/PycharmProjects/Shapley-NAS/test.py:84: UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead. target = Variable(target, volatile=True).cuda(non_blocking=True) 11/16 03:32:44 PM test 000 2.667051e-02 98.958336 100.000000 11/16 03:32:48 PM test 050 1.235316e-01 96.854576 99.938728 11/16 03:32:51 PM test 100 1.133232e-01 96.967827 99.958748 11/16 03:32:51 PM test_acc 96.970001
Thank you.
Hello,
Some errors occurred while running test.py. I changed the code to solve the error, and it runs well, leaving a message.
I've confirmed that this code change has no effect on train.py and search.py.
The error I encountered is:
utils.py 120 line
test.py 84 lines
test.py 91~93 lines
Thank you.