Skip to content

Commit

Permalink
sweep: update ckpt_path, so that it's loaded in evaluation
Browse files Browse the repository at this point in the history
  • Loading branch information
caio-freitas committed Mar 19, 2024
1 parent d742c4c commit ee1c558
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sweep.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ def train(cfg: DictConfig) -> None:
cfg.policy.lr = wandb.config.lr
cfg.policy.denoising_network.hidden_dim = wandb.config.hidden_dim
cfg.policy.denoising_network.num_layers = wandb.config.num_layers
cfg.policy.ckpt_path + f"_lr{cfg.policy.lr}_hd{wandb.config.hidden_dim}_nl{wandb.config.num_layers}"
cfg.policy.ckpt_path = cfg.policy.ckpt_path + f"_lr{cfg.policy.lr}_hd{wandb.config.hidden_dim}_nl{wandb.config.num_layers}"

# instanciate policy from cfg file
policy = hydra.utils.instantiate(cfg.policy)
Expand Down

0 comments on commit ee1c558

Please sign in to comment.