def test_dqn(): args = DQNArgs() env = gym.make(args.env_name) agent = DQNAgent(env, QNet, SimpleNormalizer, args) agent.load(args.save_dir) for _ in range(10): agent.test_one_episode(True)