search

LiuLei95 / PyTorch-Learned-Image-Compression-with-GMM-and-Attention

This repo is implementation for Learned Image Compression with Discretized Gaussian Mixture Likelihoods and Attention Modules in pytorch.
Apache License 2.0
62 stars 2 forks source link

Training error #17

Open DrCMY opened 5 months ago

DrCMY commented 5 months ago

Hi, I have no problem with inference by using the pre trained models. But training throws error for means and sigmas used by Laplace distribution. Means and sigmas outputs from entropy function often are not valid Laplace function inputs. This happens randomly at any epoch. Anyone having the same issue or has a suggestion?

songchengpeng commented 1 month ago

sigma1, sigma2, sigma3 = sigma1.clamp(1e-10, 1e10), sigma2.clamp(1e-10, 1e10), sigma3.clamp(1e-10, 1e10)