wangqingl opened this issue a year ago · 1 comments
Loading the dataset takes up too much memory, resulting in frequent training stops, 512 × The image size is 512, and the batch size is 8.
You can try load only the dataset list, and load data in each training iteration.