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
def run(hparams):
# init_experiment(comment=hparams.experiment_name)
init_experiment()
init_experiment(comment=hparams.experiment_name)
from utils import CURR_RUN_PATH, logger # import these after init_experiment
logger.info(CURR_RUN_PATH)
logger.info(f"tensorboard --logdir={CURR_RUN_PATH}")
......@@ -49,6 +48,8 @@ def run(hparams):
if __name__=='__main__':
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],
# help='how many gpus to use in the node.'
# ' value -1 uses all the gpus on the node')
......
Markdown is supported
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