notwa/smbot
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

fix antithetic ARS

Browse source
This commit is contained in:
Connor Olding 2018-06-12 20:51:08 +02:00
parent ccce6a2d55
commit 74eb2bfbef
1 changed files with 7 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
ars.lua
View file

@ -105,12 +105,13 @@ function Ars:ask(graycode)
local asking = zeros(self.dims)
local noisy = zeros(self.dims)
asked[i] = asking
noise[i] = noisy
if self.antithetic and i % 2 == 0 then
for j, v in ipairs(self._params) do
asking[i] = v - noisy[j]
local old_noisy = noise[i - 1]
for j, v in ipairs(old_noisy) do
noisy[j] = -v
end
else
if graycode ~= nil then
for j = 1, self.dims do
@ -124,13 +125,11 @@ function Ars:ask(graycode)
noisy[j] = self.sigma * nn.normal()
end
end
for j, v in ipairs(self._params) do
asking[j] = v + noisy[j]
end
end
noise[i] = noisy
for j, v in ipairs(self._params) do
asking[j] = v + noisy[j]
end
end
self.noise = noise