소스 검색

raise load error

nagadomi 9 년 전
부모
커밋
0300c714eb
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      tools/benchmark.lua

+ 1 - 0
tools/benchmark.lua

@@ -371,6 +371,7 @@ function load_noise_scale_model(model_dir, noise_level, force_cudnn)
       local scale
       s1, scale = pcall(w2nn.load_model, f, force_cudnn)
       if not s1 then
+	 error(model_dir .. ": load error")
 	 return nil
       end
       model.scale_model = scale