berenslab / retinal-rl

Testing theories about retinal coding in reinforcement learning environments
GNU Affero General Public License v3.0
0 stars 1 forks source link

Sparsity debugging #45

Closed alex404 closed 12 hours ago

alex404 commented 3 days ago

Mostly a bit of code cleanup around sparsities. KL Sparsity should really only be applied to sigmoid outputs... I could put in some kind of warning but I might deprecate it anyway because it seems rarely used.

Also added back in a shallow vs deep autoencoder, where basically the decoder is pre or post the first couple FC layers. Still not having much luck with the deep autoencoder, but now that simulation performance is way better, I should make a bigger effort.