Skip to content

关于actor_torch的预训练文件q_network.ckpt和learner_torch的save(args.ckpt_path / 'ckpt') #16

@LeslieCode7717

Description

@LeslieCode7717

请问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函数,无法保存模型,只能保存保存模型权重

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions