create a tensor proto whose content is larger than 2GB

前端 未结 2 357
清歌不尽
清歌不尽 2021-01-19 13:10

I created a ndarray (W) which size is (2^22, 256), and I tried to use this array as my initialization of weight matirx using:

w = tf.Variable(tf.convert_to_t         


        
2条回答
  •  暗喜
    暗喜 (楼主)
    2021-01-19 14:01

    for tf v1.14.0 you can solve this with tf.compat.v1.enable_eager_execution() tf v2.0+ doesn't throw error in situation at all.

提交回复
热议问题