2733bef356c53286d475a67476d88d4840923830,code/deep/finetune_AlexNet_ResNet/finetune_office31.py,,,#,15

Before Change

args = parser.parse_args()

// Parameter setting
DEVICE = torch.device("cuda:" + str(args.cuda)
                      if torch.cuda.is_available() else "cpu")
N_CLASS = 31
LEARNING_RATE = 1e-4
BATCH_SIZE = {"src": int(args.batchsize), "tar": int(args.batchsize)}

After Change

parser.add_argument("--n_epoch", type=int, default=100)
parser.add_argument("--momentum", type=float, default=0.9)
parser.add_argument("--decay", type=float, default=5e-4)
parser.add_argument("--data", type=str, default="Your dataset folder")
parser.add_argument("--early_stop", type=int, default=20)
args = parser.parse_args()

// Parameter setting

Instances