Skip to content

Commit

Permalink
Update train_reader.py
Browse files Browse the repository at this point in the history
  • Loading branch information
wangcunxiang authored Jul 15, 2023
1 parent 1c4db37 commit 38090d4
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions reader/train_reader.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,10 +163,7 @@ def get_parameters(model):
options.add_optim_options()
opt = options.parse()
#opt = options.get_options(use_reader=True, use_optim=True)
if opt.local_rank in [-1, 0] and not opt.no_wandb:
wandb.login(key="64d4aba41acda3b77b50e25d595fb18fdf327590")
wandb.init(project="FiD_struc", entity="seanwang", notes=opt.name)


print('name = {}'.format(opt.name))
torch.manual_seed(opt.seed)
src.slurm.init_distributed_mode(opt)
Expand Down Expand Up @@ -249,4 +246,4 @@ def get_parameters(model):
collator,
best_dev_em,
checkpoint_path
)
)

0 comments on commit 38090d4

Please sign in to comment.