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

Fix: Performed onplace operation on tensor requiring diff

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-08-22 01:24:40 +02:00
parent f93d9adc66
commit 5b035a4ed8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
priorConditionedAnnealing/pca.py
View File

@ -147,7 +147,7 @@ class PCA_Distribution(SB3_Distribution):
assert self.skip_conditioning or type(traj) != type(None), 'A past trajectory has to be supplied if conditinoning is performed'
pi_mean, pi_std = self.distribution.mean.cpu(), self.distribution.scale.cpu()
rho_mean, rho_std = self._conditioning_engine(traj, pi_mean, pi_std)
rho_std *= f_sigma
rho_std = rho_std * f_sigma
eta = self._get_rigged(pi_mean, pi_std,
rho_mean, rho_std,
epsilon)