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.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment