CodeMirror
/
waifu2x
mirror of
1
0
Fork 0
Code Issues Releases Wiki Activity

Use w2nn.load_model to use cudnn #369

This commit is contained in:
nagadomi 2020-10-28 09:40:40 +09:00
parent 44503fb4c0
commit c9b0fe7f41
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.lua
View File

@ -526,7 +526,7 @@ local function train()
}
local model
if settings.resume:len() > 0 then
model = torch.load(settings.resume, "ascii")
model = w2nn.load_model(settings.resume, settings.backend == "cudnn", "ascii")
adam_config.xEvalCount = math.floor((#train_x * settings.patches) / settings.batch_size) * settings.batch_size * settings.inner_epoch * (settings.resume_epoch - 1)
print(string.format("set eval count = %d", adam_config.xEvalCount))
if adam_config.xEvalCount > 0 then