File “/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py”, line 3257, in run_cell_async
File “/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py”, line 3473, in run_ast_nodes
File “/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py”, line 3553, in run_code
File “”, line 204, in <cell line: 204>
File “/usr/local/lib/python3.10/dist-packages/keras/src/utils/traceback_utils.py”, line 65, in error_handler
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py”, line 1807, in fit
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py”, line 1401, in train_function
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py”, line 1384, in step_function
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py”, line 1373, in run_step
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py”, line 1150, in train_step
File “/usr/local/lib/python3.10/dist-packages/keras/src/utils/traceback_utils.py”, line 65, in error_handler
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py”, line 590, in call
File “/usr/local/lib/python3.10/dist-packages/keras/src/utils/traceback_utils.py”, line 65, in error_handler
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/base_layer.py”, line 1149, in call
File “/usr/local/lib/python3.10/dist-packages/keras/src/utils/traceback_utils.py”, line 96, in error_handler
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/sequential.py”, line 398, in call
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/functional.py”, line 515, in call
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/functional.py”, line 672, in _run_internal_graph
File “/usr/local/lib/python3.10/dist-packages/keras/src/utils/traceback_utils.py”, line 65, in error_handler
File “/usr/local/lib/python3.10/dist-packages/keras/src/engine/base_layer.py”, line 1149, in call
File “/usr/local/lib/python3.10/dist-packages/keras/src/utils/traceback_utils.py”, line 96, in error_handler
File “/usr/local/lib/python3.10/dist-packages/keras/src/layers/normalization/batch_normalization.py”, line 597, in call
File “/usr/local/lib/python3.10/dist-packages/keras/src/layers/normalization/batch_normalization.py”, line 990, in _fused_batch_norm
File “/usr/local/lib/python3.10/dist-packages/keras/src/utils/control_flow_util.py”, line 108, in smart_cond
File “/usr/local/lib/python3.10/dist-packages/keras/src/layers/normalization/batch_normalization.py”, line 964, in _fused_batch_norm_training
OOM when allocating tensor with shape[32,128,224,224] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[{{node sequential/batch_normalization_13/FusedBatchNormV3}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. This isn’t available when running in Eager mode.
[Op:__inference_train_function_16071]