【tf.keras】Resource exhausted: OOM when allocating tensor with shape [9216,4096] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc 运行以下类似代码: whileTrue: inputs, outputs = get_AlexNet() model = tf.keras.Model(inputs=inputs, outputs=outputs) ...
【tf.keras】Resource exhausted: OOM when allocating tensor with shape [9216,4096] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc 运行以下类似代码: whileTrue: inputs, outputs = get_AlexNet() model = tf.keras.Model(inputs=inputs, outputs=outputs) ...