Pytorch-Deep-Neural-Networks
Pytorch-Deep-Neural-Networks copied to clipboard
pytorch >>> 快速搭建自己的模型!
Results
2
Pytorch-Deep-Neural-Networks issues
Sort by
recently updated
recently updated
newest added
最近用了一下DBN在CUDA上面跑代码,发现一个运行的小问题 dbn.py 里 52行 ` s = (torch.rand(p.size())< p).float().to(self.dvc)` 会报错 Traceback (most recent call last): File "mnist_cls.py", line 55, in model.run(e=3, pre_e=3) File "../core/epoch.py", line 98, in run self.pre_batch_training(pre_e, b)...
感觉这个代码写的很完整很优美,当时看了很久的readme,也不知道如何开始运行这个代码。找不到程序入口。一般程序有要给mian.py或者train.py 之类的。这个没有,都是文件夹,不知道如何下手。。。尴尬。请指教。。。