agent.py 文件源码

def save(self, dirname):
        """Save internal states."""
        makedirs(dirname, exist_ok=True)
        for attr in self.saved_attributes:
            assert hasattr(self, attr)
            attr_value = getattr(self, attr)
            if isinstance(attr_value, AttributeSavingMixin):
                assert attr_value is not self, "Avoid an infinite loop"
                attr_value.save(os.path.join(dirname, attr))
            else:
                serializers.save_npz(
                    os.path.join(dirname, '{}.npz'.format(attr)),
                    getattr(self, attr))