e6b394f7340cbeee16bf4424cdf0367f80b60619,code/deep/finetune_AlexNet_ResNet/office31.py,,,#,112

Before Change

    model = load_model(model_name).to(DEVICE)
    lrs = LEARNING_RATE
    for e in tqdm(range(1, N_EPOCH + 1)):
        tqdm.write("learning rate: " + str(lrs))
        optimizer = get_optimizer(model_name,learning_rate=lrs)
        train(e, model, optimizer, data_src)
        test(e, model, data_tar)

After Change

    // Load data
    root_dir = "data/OFFICE31/"
    domain = {"src": "amazon", "tar": "webcam"}
    dataloaders = {x: data_loader.load_data(root_dir, domain[x], BATCH_SIZE[x], x) for x in ["src", "tar"]}
    // Load model
    model_name = str(args.model)
    model = load_model(model_name).to(DEVICE)
    print("Source:{}, target:{}, model: {}".format(domain["src"], domain["tar"], model_name))

Instances