What mistake m i making in adding new weight Layer

https://www.kaggle.com/jaideepvalani/arcface-humpback-customhead-fastai

Hi all
please have a look at this kernel…
This is leading into unstable loss and accuracy…Some times 0 accurcy some times its more. Check the training output at the end
I suspect i have not added weights with Arc layer properly