Hello @hhk7734
On training the model on yolov4 and training it using yolov4.conv.137 weights file, I get an OOM.
batch_size = 8
input_size = 416
I am using 2.1.0 version of yolov4 from this repo.
I have changed
physical_devices = tf.config.experimental.list_physical_devices("GPU")
if len(physical_devices) > 0:
tf.config.experimental.set_memory_growth(physical_devices[0], True)
tf.config.gpu.set_memory_growth(physical_devices[0],True)
Hello @hhk7734 On training the model on yolov4 and training it using yolov4.conv.137 weights file, I get an OOM. batch_size = 8 input_size = 416 I am using 2.1.0 version of yolov4 from this repo.
I have changed
to
as set_memory_growth doesn't work.
following are the relevant dependencies.
tensorflow version is 2.2.0 cuda 10.2 cudnn 7.6.1
Here is the training script:
Am I doing something wrong?