About the './dmp/data/pretrained_weights.pt' file

shikharbahl / neural-dynamic-policies

MIT License

63 stars 13 forks source link

About the './dmp/data/pretrained_weights.pt' file #7

Closed linksdl closed 3 years ago

linksdl commented 3 years ago

Hi, @shikharbahl, I am struggling with a problem in 'main_il.py'. Is this parameter pt='./dmp/data/pretrained_weights.pt your trained model (ndp_cnn) ? If yes, could you upload this file, please?

shikharbahl commented 3 years ago

Unfortunately, I lost have acess to the pretrained weights, but the network was trained on MNIST: https://github.com/pytorch/examples/blob/master/mnist/main.py.