def main(): global args args = parser.parse_args() if args.nn == "densenet10" or args.nn == "wideresnet10": c.test(args.nn, args.out_dataset, args.gpu, args.magnitude, args.temperature) else: cv.test(args.nn, args.out_dataset, args.gpu, args.magnitude, args.temperature)
def main(): global args args = parser.parse_args() if args.version == "test": c.test(args.nn, args.out_dataset, args.magnitude, args.temperature) else: c.train() # densenet training
def main(): print("Start") global args args = parser.parse_args() os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu c.test(args.in_dataset, args.out_dataset, args.wide, args.magnitude, args.temperature)
def main(): global args args = parser.parse_args() c.test(args.nn, args.out_dataset, args.gpu, args.magnitude, args.temperature)