Fix model name

Author: laurens

main.py

@@ -84,14 +84,14 @@ def run(args):
         if args.switch_goal and logger.n_eps == 1000:
             torch.save({'model_params': option_critic.state_dict(),
                         'goal_state': env.goal},
-                        'models/option_critic_{args.seed}_1k')
+                        f'models/option_critic_seed={args.seed}_1k')
             env.switch_goal()
             print(f"New goal {env.goal}")
 
         if args.switch_goal and logger.n_eps > 2000:
             torch.save({'model_params': option_critic.state_dict(),
                         'goal_state': env.goal},
-                        'models/option_critic_{args.seed}_2k')
+                        f'models/option_critic_seed={args.seed}_2k')
             break
 
         done = False ; ep_steps = 0 ; option_termination = True ; curr_op_len = 0