From c9b0fe7f411bc0b92b852cad4fef4f22f501684d Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Wed, 28 Oct 2020 09:40:40 +0900
Subject: [PATCH] Use w2nn.load_model to use cudnn #369

---
 train.lua | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.lua b/train.lua
index f12ffa6..7f65ad2 100644
--- a/train.lua
+++ b/train.lua
@@ -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