fast-neural-style
fast-neural-style copied to clipboard
Training killed
Hi Justin:
I was training with the following command:
th train.lua -h5_file dataset_file.h5 -style_image images/styles/the_scream.jpg -style_image_size 384 -content_weights 1.0 -style_weights 5.0 -checkpoint_name checkpoint -gpu -1
The program started and printed out the network, then it said "Killed" and exited. I am wondering if you have experienced the same problem before. Is it due to memory issue or something else?
Here is the print-out:
Initializing model from scratch nn.Sequential { [input -> (1) -> (2) -> (3) -> (4) -> (5) -> (6) -> (7) -> (8) -> (9) -> (10) -> (11) -> (12) -> (13) -> (14) -> (15) -> (16) -> (17) -> (18) -> (19) -> (20) -> (21) -> (22) -> (23) -> (24) -> output]
///network details
Killed.
Thanks a lot.
It seems like you don't have enough memory if you working on -gpu -1
. Try to reduce the image_size may help :)
I am pretty sure that you don't have enough memor. you can use swap partition