Pytorch-Deep-Neural-Networks icon indicating copy to clipboard operation
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 之类的。这个没有,都是文件夹,不知道如何下手。。。尴尬。请指教。。。