xyxue1122

CPU memory leak when training with CsvDataset-like dataset

10

Hi, I tried to train with siglip loss on a large dataset and I found that during training (not evaluation), CPU memory usage kept increasing. The program was finally killed...