Commit f3f01a50 authored by Verena Praher's avatar Verena Praher
Browse files

add experiment_name to resnets exp

parent efa025cd
...@@ -7,8 +7,7 @@ import os ...@@ -7,8 +7,7 @@ import os
def run(hparams): def run(hparams):
# init_experiment(comment=hparams.experiment_name) init_experiment(comment=hparams.experiment_name)
init_experiment()
from utils import CURR_RUN_PATH, logger # import these after init_experiment from utils import CURR_RUN_PATH, logger # import these after init_experiment
logger.info(CURR_RUN_PATH) logger.info(CURR_RUN_PATH)
logger.info(f"tensorboard --logdir={CURR_RUN_PATH}") logger.info(f"tensorboard --logdir={CURR_RUN_PATH}")
...@@ -49,6 +48,8 @@ def run(hparams): ...@@ -49,6 +48,8 @@ def run(hparams):
if __name__=='__main__': if __name__=='__main__':
parent_parser = HyperOptArgumentParser(strategy='grid_search', add_help=False) parent_parser = HyperOptArgumentParser(strategy='grid_search', add_help=False)
parent_parser.add_argument('--experiment_name', type=str,
default='pt_lightning_exp_a', help='test tube exp name')
#parent_parser.add_argument('--gpus', type=list, default=[0,1], #parent_parser.add_argument('--gpus', type=list, default=[0,1],
# help='how many gpus to use in the node.' # help='how many gpus to use in the node.'
# ' value -1 uses all the gpus on the node') # ' value -1 uses all the gpus on the node')
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment