[Feat] working sinkhorn version and gradient checkpointing in batch dimension#15

Open

SCMusson wants to merge 4 commits intoDegiacomi-Lab:masterfrom SCMusson:feat_sinkhorn

+496-243

Commits on Oct 26, 2023

moved current broken sinkhorn to deprecated/sinkhorn_trainer.py
Samuel Musson
committed
add functional sinkhorn trainer
Samuel Musson
committed

Commits on Oct 27, 2023

bug fix for sinkhorn_trainer and added gradient checkpointing it utils
Samuel Musson
committed
sinkhorn example script and gradient checkpointing example script
Samuel Musson
committed