fast-neural-style icon indicating copy to clipboard operation
fast-neural-style copied to clipboard

Training killed

Open gecong opened this issue 8 years ago • 2 comments

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.

gecong avatar Oct 10 '16 03:10 gecong

It seems like you don't have enough memory if you working on -gpu -1. Try to reduce the image_size may help :)

SilvesterHsu avatar Oct 14 '16 09:10 SilvesterHsu

I am pretty sure that you don't have enough memor. you can use swap partition

WestCoastBaby avatar Apr 05 '17 07:04 WestCoastBaby