Just to inform, I'm working in optimus to develop python code to train the neural network since it requires a lot of memory.