예제 #1
0
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)
예제 #2
0
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
예제 #3
0
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)