1
0
Fork 0
mirror of synced 2024-04-30 02:52:26 +12:00

Merge branch 'master' of github.com:nagadomi/waifu2x

This commit is contained in:
nagadomi 2020-01-10 05:09:47 +09:00
commit da44c59fb0

View file

@ -12,8 +12,8 @@ local function create_filters(ch, n, k, layers)
if i == 1 then
-- channel identity
for j = 1, ch do
filter.weight[i]:fill(0)
filter.weight[i][i][math.floor(k/2)+1][math.floor(k/2)+1] = 1
filter.weight[j]:fill(0)
filter.weight[j][j][math.floor(k/2)+1][math.floor(k/2)+1] = 1
end
end
model:add(filter)