VanillaNet icon indicating copy to clipboard operation
VanillaNet copied to clipboard

eval模式build_dataset时的错误

Open laozizhenniubi opened this issue 8 months ago • 1 comments

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

laozizhenniubi avatar Jun 17 '24 09:06 laozizhenniubi