I'm using the xavier conv2d method for initializing my variables like this:
initializer = tf.contrib.layers.xavier_initializer_conv2d()
variable = tf.get_variable(name=name,shape=shape,initializer=initializer)
the training runs locally using gcloud ml-engine local train
, however it crashes when sending it as a job to the cloud.
The crash log: "Module raised an exception <type 'exceptions.SystemExit'>:-15."
If I replace the xavier initializer by a random uniform initializer, the training works both on my local machine and on the cloud:
initializer = tf.random_uniform_initializer(-0.25,0.25)
I'm running gpu_enabled tensorflow version 1.01 on my local machine using python 2.7.13