Closed XiaoHao-Chen closed 5 years ago
What are the formatting requirements for CIFAR and mnist datasets
Thanks you code firstly. I want to know how much GPU you need to train, what version of Python and Pytorch. When I run main.py, there is a hint of: KeyError: ' RUNS\\PCNN_LR '. But I don't know anything about the mistake. Could you give me some advice?Thank you
Hi,
you will need pytorch 0.3 or 0.4 (I think it was originally coded on 0.3, but should be 0.4 compatible). The dataset formatting requirements are already included in the repository. As for hardware, to train the full model you will about 4 GPUs for a week. You can also train a smaller version in less than a day on a single GPU and get competitive performance.
For the last error, could you provide more information ? a full stack trace would help.
-Lucas
嗨,
你需要0.3或0.4的pytorch(我认为它最初编码为0.3,但应该是0.4兼容)。数据集格式化要求已包含在存储库中。至于硬件,要训练完整的模型,您将在一周内获得大约4个GPU。您还可以在一个GPU上在不到一天的时间内训练较小的版本并获得有竞争力的性能。
对于最后一个错误,您能提供更多信息吗?完整的堆栈跟踪会有所帮助。
-Lucas
Thank you for your reply. I have solved the question I raised. Another question,:I see the trained model you released, how do I use these models? How to use the trained model directly to complete the main work of PIXELCNN?
I really Windows 7 running your code under the system. After downloading the good data and after, the following error occurred:
Traceback (most recent call last):
File "D:/pixel-cnn-pp-master/main.py", line 119, in
This probably means that you are not using fork to start your
child processes and you have forgotten to use the proper idiom
in the main module:
if __name__ == '__main__':
freeze_support()
I found some solutions on the internet, but it didn't work. So can you give me some advice?Thank you!
And after the above error occurred, the program did not stop, but continued to run, but nothing came of it. Why is that?
Sometimes the pytorch dataloaders throws errors and keeps running. Usually this is fine. Try playing with the num_workers
flag, maybe it will help.
Sorry I don't have a better answer for you
Thank you very much for your help!
How do you use this code, please? Including the reading of datasets, the configuration of the environment, hardware requirements, and so on? Thank you for your help.