How do I make my own data set for training

[cyt520]

Hi Author, your work has inspired me a lot and I would like to ask how I can make my own data set to train this model。

[Lucky-xiaoxuan24]

你好，请问你配置环境的时候python setup.py build develop解决了吗

[cyt520]

解决了，安装对应的torch和cuda版本就好了

[cuishuhuan]

能问问您安装的哪个版本的torch和CUDA吗

[cyt520]

您可以询问安装哪个版本的 torch 和 CUDA 吗？
cuda 11.0 torch1.7.1+cu110

[jun3333333]

Have you prepared your dataset?I'd like to consult you about knowledge related to datasets. My email address is 2373178085@qq.com.