Skip to content

Commit e886a40

Browse files
committed
resolve the sweep issue
1 parent e1b95cb commit e886a40

File tree

4 files changed

+9
-8
lines changed

4 files changed

+9
-8
lines changed

main.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
# 1- Change the naming of the functions ---> ( Done)
4545
# 2- Change the algorithms following DCORAL --> (Done)
4646
# 3- Keep one trainer for both train and test -->(Done)
47-
# 4- Creat the new joint loader that consider the all possible batches --> Done
48-
# 4- Add the best hparams
49-
# 5- Implement Lower/Upper Bound Approach
50-
# 4- Add pretrain based methods (ADDA, MCD)
47+
# 4- Create the new joint loader that consider the all possible batches --> Done
48+
# 5- Implement Lower/Upper Bound Approach --> Done
49+
# 6- Add the best hparams --> Done
50+
# 7- Add pretrain based methods (ADDA, MCD, MDD)

main_sweep.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -11,15 +11,16 @@
1111
help='DANN, Deep_Coral, WDGRL, MMDA, VADA, DIRT, CDAN, ADDA, HoMM, CoDATS')
1212

1313
# ========= Select the DATASET ==============
14-
parser.add_argument('--data_path', default=r'../data', type=str, help='Path containing datase2t')
14+
parser.add_argument('--data_path', default=r'../ADATIME_data', type=str, help='Path containing datase2t')
1515
parser.add_argument('--dataset', default='HAR', type=str, help='Dataset of choice: (WISDM - EEG - HAR - HHAR_SA)')
1616

1717
# ========= Select the BACKBONE ==============
1818
parser.add_argument('--backbone', default='CNN', type=str, help='Backbone of choice: (CNN - RESNET18 - TCN)')
1919

2020
# ========= Experiment settings ===============
21-
parser.add_argument('--num_runs', default=3, type=int, help='Number of consecutive run with different seeds')
21+
parser.add_argument('--num_runs', default=1, type=int, help='Number of consecutive run with different seeds')
2222
parser.add_argument('--device', default="cuda", type=str, help='cpu or cuda')
23+
parser.add_argument('--exp_name', default='sweep_EXP1', type=str, help='experiment name')
2324

2425
# ======== sweep settings =====================
2526
parser.add_argument('--num_sweeps', default=1, type=str, help='Number of sweep runs')

trainers/abstract_trainer.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def __init__(self, args):
3636
# Exp Description
3737
self.experiment_description = args.dataset
3838
self.run_description = f"{args.da_method}_{args.exp_name}"
39-
39+
4040
# paths
4141
self.home_path = os.getcwd() #os.path.dirname(os.getcwd())
4242
self.save_dir = args.save_dir

trainers/sweep.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from models.models import get_backbone_class
2121
from utils import AverageMeter
2222

23-
from abstract_trainer import AbstractTrainer
23+
from trainers.abstract_trainer import AbstractTrainer
2424

2525
warnings.filterwarnings("ignore", category=sklearn.exceptions.UndefinedMetricWarning)
2626
parser = argparse.ArgumentParser()

0 commit comments

Comments
 (0)