VanillaNet
VanillaNet copied to clipboard
eval模式build_dataset时的错误
约229行,在源代码的main中,
dataset_train, args.nb_classes = build_dataset(is_train=True, args=args)
没有考虑到仅eval模式,is_train默认是true,应该在前加一个条件判断语句。