Skip to content
Snippets Groups Projects
Commit 60f8dcb8 authored by Schneider Leo's avatar Schneider Leo
Browse files

add : wandb random name

parent 956b97c4
No related branches found
No related tags found
No related merge requests found
......@@ -102,6 +102,9 @@ def run_duo(args):
os.environ["WANDB_DIR"] = os.path.abspath("./wandb_run")
wdb.init(project="contrastive_classification", dir='./wandb_run')
wdb.log({'hparam/optimizer':args.opti,'hparam/p_prop':args.positive_prop,'hparam/sampler':args.sampler,
'hparam/lr':args.lr,'hparam/data_train':args.dataset_train_dir,'hparam/data_val':args.dataset_train_dir,
'hparam/noise':args.noise_threshold,'hparam/model':args.model})
print('Wandb initialised')
# load data
data_train, data_val_batch, data_test_batch = load_data_duo(base_dir_train=args.dataset_train_dir,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment