Error2

[Vosatorp]

command: python singing_transcription.py -i ../audio/test.wav -o ../output

2024-06-08 17:46:28.019747: W tensorflow/core/framework/op_kernel.cc:1767] OP_REQUIRES failed at conv_ops.cc:662 : Resource exhausted: OOM when allocating tensor with shape[64,64,31,513] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
Traceback (most recent call last):
  File "singing_transcription.py", line 131, in <module>
    main(parser.parse_args())
  File "singing_transcription.py", line 81, in main
    fl_note = ST.predict_melody(model_ST, path_audio)  # frame-level pitch score                                                  
  File "singing_transcription.py", line 38, in predict_melody
    y_predict = model_ST.predict(X_test, batch_size=self.batch_size, verbose=1)                                                   
  File "/usr/local/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 1727, in predict                 
    tmp_batch_outputs = self.predict_function(iterator)
  File "/usr/local/lib/python3.8/site-packages/tensorflow/python/eager/def_function.py", line 889, in __call__                    
    result = self._call(*args, **kwds)
  File "/usr/local/lib/python3.8/site-packages/tensorflow/python/eager/def_function.py", line 956, in _call                       
    return self._concrete_stateful_fn._call_flat(
  File "/usr/local/lib/python3.8/site-packages/tensorflow/python/eager/function.py", line 1960, in _call_flat                     
    return self._build_call_outputs(self._inference_function.call(                                                                
  File "/usr/local/lib/python3.8/site-packages/tensorflow/python/eager/function.py", line 591, in call                            
    outputs = execute.execute(
  File "/usr/local/lib/python3.8/site-packages/tensorflow/python/eager/execute.py", line 59, in quick_execute                     
    tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name,                                                        
tensorflow.python.framework.errors_impl.ResourceExhaustedError:  OOM when allocating tensor with shape[64,64,31,513] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc                                                           
         [[node model/conv_s1_1x1/Conv2D (defined at singing_transcription.py:38) ]]                                              
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.
 [Op:__inference_predict_function_5418]

Function call stack:
predict_function