-
Notifications
You must be signed in to change notification settings - Fork 14
Open
Labels
Description
We want to extend our model training package to support advanced data augmentation techniques, specifically Mix-Up. The goal is to create a wrapper or class (MixUpModel) that allows users to train any model using the Mix-Up method.
Requirements:
- The model should accept any base architecture (CNN, MLP, etc.).
- During training, inputs and labels should be mixed according to the Mix-Up technique.
- The user should be able to toggle Mix-Up on/off via a parameter.
- Ensure compatibility with standard TensorFlow/Keras training loops.
- Optional: Integrate with ensemble or transformer-based preprocessing pipelines in the future.
Reactions are currently unavailable