Skip to content

overwrite training_step for CustomDPOTrainer to clear cuda cache every train step#5019

Open
zzc0430 wants to merge 1 commit intohiyouga:mainfrom zzc0430:feature/dpo_empty_cache
+10-2

Commits

Commits on Jul 30, 2024