Fix validation metric
This commit is contained in:
parent
a92b2cb386
commit
fa9355be7c
|
@ -48,6 +48,12 @@ local function make_validation_set(x, transformer, n, patches)
|
||||||
xlua.progress(i, #x)
|
xlua.progress(i, #x)
|
||||||
collectgarbage()
|
collectgarbage()
|
||||||
end
|
end
|
||||||
|
local new_data = {}
|
||||||
|
local perm = torch.randperm(#data)
|
||||||
|
for i = 1, perm:size(1) do
|
||||||
|
new_data[i] = data[perm[i]]
|
||||||
|
end
|
||||||
|
data = new_data
|
||||||
return data
|
return data
|
||||||
end
|
end
|
||||||
local function validate(model, criterion, data, batch_size)
|
local function validate(model, criterion, data, batch_size)
|
||||||
|
|
Loading…
Reference in a new issue