1
0
Fork 0
mirror of synced 2024-05-06 05:52:22 +12:00

Fix noise0 model (backend: cudnn -> cunn)

This commit is contained in:
nagadomi 2016-08-07 01:41:42 +09:00
parent c759f040bc
commit 7040dff6f8
5 changed files with 1 additions and 1 deletions

View file

@ -21,7 +21,7 @@ rebuild() {
fi
if [ -f models/${1}/noise0_scale2.0x_model.t7 ] && [ ! -h models/${1}/noise0_scale2.0x_model.t7 ]; then
th tools/rebuild_model.lua -i models/${1}/noise1_scale2.0x_model.t7 -o models/${1}/noise1_scale2.0x_model.t7 -backend cunn -model $2
th tools/rebuild_model.lua -i models/${1}/noise0_scale2.0x_model.t7 -o models/${1}/noise0_scale2.0x_model.t7 -backend cunn -model $2
fi
if [ -f models/${1}/noise1_scale2.0x_model.t7 ] && [ ! -h models/${1}/noise1_scale2.0x_model.t7 ]; then
th tools/rebuild_model.lua -i models/${1}/noise1_scale2.0x_model.t7 -o models/${1}/noise1_scale2.0x_model.t7 -backend cunn -model $2