From 4cf707e79c508db6a5e9f28faab936c7746cd9ac Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 26 Nov 2018 14:27:08 +0100 Subject: continue train --- app/relay/modules/pix2pixhd.js | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'app/relay') diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js index a01e321..db40781 100644 --- a/app/relay/modules/pix2pixhd.js +++ b/app/relay/modules/pix2pixhd.js @@ -49,8 +49,9 @@ const train = { const datasets_path = path.join(cwd, 'datasets', dataset) const checkpoints_path = path.join(cwd, 'checkpoints', dataset) const iter_txt = path.join(checkpoints_path, 'iter.txt') + const checkpoint_path = path.join(checkpoints_path, 'latest_net_G.pth') console.log(dataset, iter_txt) - if (fs.existsSync(iter_txt)) { + if (fs.existsSync(iter_txt) && fs.existsSync(checkpoint_path)) { const iter = fs.readFileSync(iter_txt).toString().split('\n'); console.log(iter) epoch = iter[0] || 0 -- cgit v1.2.3-70-g09d2