In this publication titled bert pytorch classification example, I have provided a complete image from start to finish ,Please look below
