dodox/PriorConditionedAnnealing
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Performance Optimizations when skip_conditioning=True

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-05-21 17:37:45 +02:00
parent 4ba7db831f
commit 35df8f44da
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
priorConditionedAnnealing/pca.py
View File

@ -183,6 +183,9 @@ class PCA_Distribution(SB3_Distribution):
return traj[:, -self.window:, :] return traj[:, -self.window:, :]
def _conditioning_engine(self, trajectory, pi_mean, pi_std): def _conditioning_engine(self, trajectory, pi_mean, pi_std):
if self.skip_conditioning:
return pi_mean, pi_std
traj = self._pad_and_cut_trajectory(trajectory) traj = self._pad_and_cut_trajectory(trajectory)
y_np = np.append(np.swapaxes(traj, -1, -2), y_np = np.append(np.swapaxes(traj, -1, -2),
np.expand_dims(pi_mean, -1), -1) np.expand_dims(pi_mean, -1), -1)