Update train_v2.py: add --dataset flag to switch between old proxy data (dataset_real) and new built dataset (dataset_v2) 61c951d verified Ellaft commited on 12 days ago
Add train_v2.py: OGM-GE training loop + asymmetric LRs + aux loss logging 0c3f8a7 verified Ellaft commited on 12 days ago