From f41466b2d5c2ef6ce991f28dc7a29b4562421477 Mon Sep 17 00:00:00 2001 From: wogong Date: Fri, 25 May 2018 00:29:14 +0800 Subject: [PATCH] add office task file. --- main_office.py | 45 ++++++++++++++++++++++++++++++++++++++++++--- 1 file changed, 42 insertions(+), 3 deletions(-) diff --git a/main_office.py b/main_office.py index c3b5d42..fec4d7a 100644 --- a/main_office.py +++ b/main_office.py @@ -1,12 +1,51 @@ -from models.model import SVHNmodel, Classifier - +import os from core.dann import train_dann from core.test import eval from models.model import AlexModel -import params from utils import get_data_loader, init_model, init_random_seed + +class Config(object): + # params for path + dataset_root = os.path.expanduser(os.path.join('~', 'Datasets')) + model_root = os.path.expanduser(os.path.join('~', 'Models', 'pytorch-DANN')) + + # params for datasets and data loader + batch_size = 128 + + # params for source dataset + src_dataset = "amazon31" + src_model_trained = True + src_classifier_restore = os.path.join(model_root, src_dataset + '-source-classifier-final.pt') + + # params for target dataset + tgt_dataset = "webcam31" + tgt_model_trained = True + dann_restore = os.path.join(model_root, src_dataset + '-' + tgt_dataset + '-dann-final.pt') + + # params for pretrain + num_epochs_src = 100 + log_step_src = 10 + save_step_src = 50 + eval_step_src = 20 + + # params for training dann + + ## for office + num_epochs = 1000 + log_step = 10 # iters + save_step = 500 + eval_step = 5 # epochs + + manual_seed = 8888 + alpha = 0 + + # params for optimizing models + lr = 2e-4 + +params = Config() + # init random seed init_random_seed(params.manual_seed)