Punit Kaushik comments

Repositories
Issues
Comments

Results 4 comments of


                                            Punit Kaushik

Resource exhausted: OOM when allocating tensor with shape[32,960,10,10]

I was trying to train an auto encoder and encountered the same error even with smaller batches, but prefetching the data made the error go away . 😄

Adding LSTM from Scratch

:-]

Test failing with tensorflow 2.17

Use TensorFlow 2.15.0 or below, as far as I am aware support for 2.15.0 > is not available.

i want to train my dataset with yolov4 and MobileNetv3 as backbone can somebody give me the cfg file for that?

I just went thru all the layers and activations that are present in the Darknet framework and the proposed h-sigmoid and h-swish activations and Adaptive Avg Pooling layer are not...