model.py 文件源码

def save(self):
        dir = "model"
        serializers.save_hdf5(dir + "/bddqn_shared_fc.model", self.shared_fc)
        serializers.save_hdf5(dir + "/bddqn_shared_fc.optimizer", self.optimizer_shared_fc)
        serializers.save_hdf5(dir + "/bddqn_head_fc.model", self.head_fc)
        serializers.save_hdf5(dir + "/bddqn_head_fc.optimizer", self.optimizer_head_fc)