Train model takes too many hours, how to optimize it.

[tangguihua]

Hello Julian, thanks for your great work to predict cancer. Here I have meet a problem when I want to study your project. I use all the data(ndsb and Luna) to re-train the model by using your code. while the ETA shows it takes about 244 hours. My configuration is interal storage: 251G and GPU storage is 12G. how can I reduce the time to train the model, can you give me some advice. Thanks.

[juliandewit]

hello, I noticed that newer keras versions have different parameters for the "fit" function. Where I gave the number of training examples as parameter, I believe it's now the number of batches.. Hence the huge train time.. You have to look it up yourself but it's somewhere in this issues list.