search

theislab / trVAE

Conditional out-of-distribution prediction
MIT License
53 stars 11 forks source link

No activation used in decoder #bug #8

Closed alirezaomidi closed 4 years ago

alirezaomidi commented 4 years ago

It seems there is a bug in the decoder: Except for the first layer (the MMD layer), no activation function is used for the layers.

https://github.com/theislab/trVAE/blob/d713be5333b4f0868bc772f31978405e0cedbe9d/trvae/models/_trvae.py#L154-L158

Naghipourfar commented 4 years ago

Hi Alireza,

Thanks for reporting. LeakyReLU activations are being used in decoder's all layers now.