vae with temporal matching with positive pair - #339

Open

edyoshikun wants to merge 11 commits into

mainfrom

Commits on Nov 10, 2025

vae with temporal matching with positive pair
edyoshikun
committed
cd4a4cc
mistake on the forward method for postiive sample
edyoshikun
committed
b8ef058
convert to float() to support the BFFloat32
edyoshikun
committed
5ff6042

Commits on Nov 13, 2025

add graph for beta vae
edyoshikun
committed
b1a9fc7
modify the code so that we have this symmetric training running forward on the anchor and positive as well as KL. The temporal loss now shouldnt fight the KL divergence
edyoshikun
committed
49b75ad

Commits on Nov 19, 2025

fix temporal matching loss to use mu instead of stochastic z
edyoshikun
committed
24749fc
batched transforms for rot90
edyoshikun
committed
4631d0d

Commits on Nov 24, 2025

go back to mean reduction.
edyoshikun
committed
a741ce7

Commits on Dec 9, 2025

Merge branch 'main' into temporal_vae
edyoshikun
committed
d71b998
modify prediction writer for vae
edyoshikun
committed
0c37679
vae prediction visualizer
edyoshikun
committed
6df108b