在谷歌云机器学习上使用xavier初始化运行作业时出错

时间:2017-03-20 15:01:23

标签: google-cloud-ml google-cloud-ml-engine

我正在使用xavier conv2d方法初始化我的变量,如下所示:

initializer = tf.contrib.layers.xavier_initializer_conv2d()
variable = tf.get_variable(name=name,shape=shape,initializer=initializer)

培训使用gcloud ml-engine local train在本地运行,但在将其作为作业发送到云时会崩溃。

崩溃日志:"Module raised an exception <type 'exceptions.SystemExit'>:-15."

如果我用随机统一初始化程序替换xavier初始化程序,那么培训在本地计算机和云计算上都有效:

 initializer = tf.random_uniform_initializer(-0.25,0.25)

我在本地计算机上使用python 2.7.13运行gpu_enabled tensorflow版本1.01

0 个答案:

没有答案