请问actor_torch的预训练文件q_network.ckpt是怎么生成的,现在我想减少状态向量的维度,但是q_network.ckpt怎么改成对应的维度?
learner_torch的learner.py脚本:
判断是否需要保存模型
if model_save_freq%args.ckpt_save_freq == 0:
if args.ckpt_save_type == 'checkpoint':
# 保存所有训练状态,包括优化器等
agent.save(args.ckpt_path / 'ckpt')
elif args.ckpt_save_type == 'weight':
# 仅保存模型权重
with open(args.ckpt_path / f'ppo{model_save_freq}.pth', 'wb') as f:
pickle.dump(weights, f)
其中agent.save(args.ckpt_path / 'ckpt')github仓库中没有save函数,无法保存模型,只能保存保存模型权重