i am trying to train tf_efficientnet_b0_ns model from popular timm repository for kaggle g2net gravitation wave detection challenge : G2Net Gravitational Wave Detection | Kaggle
during training i checked nvidia-smi and i was using around 19gb vram out of 24(i am using rtx 3090)
my problem is during 7th epoch computer freezes and i am forced to restart the pc.
here is the code and log attached : torch_freezing_pc - Pastebin.com
how to troubleshoot this annoying freezing issue?
thank you