fed75c6c752c1f54f182545798106644ee1ba6af,rlkit/core/logging.py,Logger,save_itr_params,#Logger#,279

Before Change

            elif self._snapshot_mode == "gap":
                if itr % self._snapshot_gap == 0:
                    file_name = osp.join(self._snapshot_dir, "itr_%d.pkl" % itr)
                    pickle.dump(params, open(file_name, "wb"))
            elif self._snapshot_mode == "gap_and_last":
                if itr % self._snapshot_gap == 0:
                    file_name = osp.join(self._snapshot_dir, "itr_%d.pkl" % itr)

After Change

            elif self._snapshot_mode == "gap_and_last":
                if itr % self._snapshot_gap == 0:
                    file_name = osp.join(self._snapshot_dir, "itr_%d.pkl" % itr)
                    torch.save(params, file_name)
                file_name = osp.join(self._snapshot_dir, "params.pkl")
                torch.save(params, file_name)
            elif self._snapshot_mode == "none":
                pass

Instances