holy-edge
holy-edge copied to clipboard
train not working
Hi, I was playing around with different image_width and height values for training and after even setting it back to 480 and 480 train functionality is not working anymore. I am getting the following error.
[10 Jan 2018 10h30m46s][INFO] Model weights loaded from vgg16.npy
[10 Jan 2018 10h30m46s][INFO] Added CONV-BLOCK-1+SIDE-1
[10 Jan 2018 10h30m46s][INFO] Added CONV-BLOCK-2+SIDE-2
[10 Jan 2018 10h30m46s][INFO] Added CONV-BLOCK-3+SIDE-3
[10 Jan 2018 10h30m46s][INFO] Added CONV-BLOCK-4+SIDE-4
[10 Jan 2018 10h30m46s][INFO] Added CONV-BLOCK-5+SIDE-5
[10 Jan 2018 10h30m46s][INFO] Added FUSE layer
[10 Jan 2018 10h30m46s][INFO] Build model finished: 0.1343s
[10 Jan 2018 10h30m46s][INFO] Done initializing VGG-16 model
[10 Jan 2018 10h30m47s][INFO] Training data set-up from /home/pchaudha/hed/hed-data/HED-BSDS/train_pair.lst
[10 Jan 2018 10h30m47s][INFO] Training samples 23040
[10 Jan 2018 10h30m47s][INFO] Validation samples 5760
[10 Jan 2018 10h30m47s][WARNING] Deep supervision application set to True
Traceback (most recent call last):
File "run-hed.py", line 64, in
Does anyone know why is this happening? Thank you!