This is the implementation of Online Task-Free Continual Generative and Discriminative Learning via Dynamic Cluster Memory - DCM/GraphMemory_JS_FFHQ256.py at main · dtuzi123/DCM
times[i] = t beta = _extract_into_tensor(TSFramework.teacherArray[0].diffusion.sqrt_alphas_cumprod, times, batch.shape) batch = batch * beta batchReco = batchReco * beta q_z1 = td.normal.Normal(batch, std) q_z2 = td.normal.Normal(batchReco, std) score11 = td.kl_divergence(q_...