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

fix : wandb log barlow

parent 3b2a5d7f
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
barlow_twin_like/main.py
+ 2
− 1
View file @ 86417c93
...@@ -181,7 +181,8 @@ def run(): ...@@ -181,7 +181,8 @@ def run():
'hparam/lr': args.lr, 'hparam/data_train': args.dataset_train_dir, 'hparam/lr': args.lr, 'hparam/data_train': args.dataset_train_dir,
'hparam/data_val': args.dataset_train_dir, 'hparam/data_val': args.dataset_train_dir,
'hparam/model': args.model, 'hparam/model': args.model,
'hparam/ref_dir': args.dataset_ref_dir}) 'hparam/ref_dir': args.dataset_ref_dir,
'hparam/batch': args.batch_size})
print('Wandb initialised') print('Wandb initialised')
# load data # load data
......
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