Closed fhrzn closed 2 years ago
Hello again! Thank you for the great code. As stated in the paper, there is a KL Thresholding technique. But, I feel lost looking for this function in the code. Could you please point out where I could find it?
Thank you very much!
Hi! T5VAE has a attribute min_z that it uses to enforce KL thresholding on this line.
T5VAE
min_z
Got it, thank you very much! 👍
Hello again! Thank you for the great code. As stated in the paper, there is a KL Thresholding technique. But, I feel lost looking for this function in the code. Could you please point out where I could find it?
Thank you very much!