araffin / srl-zoo

State Representation Learning (SRL) zoo with PyTorch - Part of S-RL Toolbox
https://srl-zoo.readthedocs.io/en/latest/
MIT License
162 stars 18 forks source link

Baselines #9

Closed hill-a closed 6 years ago

hill-a commented 6 years ago

Added pytorch implementation of Variational autoencoder

araffin commented 6 years ago

Also, please integrate your work in the pipeline (see pipeline.py)

araffin commented 6 years ago

I added some comments and did some clean up. Merging ;)