Spaces:
Running
Running
Fix: prevent exponential growth in augmentation
#2
by
KyungsuKim
- opened
When applying augmentation, the code concatenated y repeatedly, causing the array size doubled every iteration.
Now a separate variable y_augmented is used, so memory usage only grows linearly with the number of augmentations.
Note: I haven’t run a full test yet, so additional verification would be appreciated.