search

mfbalin / Concrete-Autoencoders

116 stars 31 forks source link

Supervised concrete autoencoder #15

Open annalauralerede opened 2 years ago

annalauralerede commented 2 years ago

Hi,

I am looking through your code and I see that for the supersived implementation you use the MSE loss instead of the cross-entropy loss like you mentioned in the paper... There is no condition to use the cross-entropy loss when supervised and it always use the mse loss. Is it a mistake or is it done on purpose?

mfbalin commented 1 year ago

One is free to use any kind of loss function for supervised training. It might be an oversight in the implementation.