astropic

**Describe the Bug** When I train a NER Flair model with APEX, the GPU memory keeps going higher until runs out. The same behaviour doesn't happen when I don't use...