koba-jon / pytorch_cpp

Deep Learning sample programs using PyTorch in C++
MIT License
241 stars 52 forks source link

About AE1d #13

Open faranaziz opened 3 years ago

faranaziz commented 3 years ago

Hello, No can find the RBM code, you not use it ad in original paper?

koba-jon commented 3 years ago

Hello, I certainly don't implement RBM in this repository (i.e., no pre-training), because I don't fully understand its algorithm. It's reported that pre-training using RBM is important in original paper. So I would like to add its functions in the future.