ClassSR
ClassSR copied to clipboard
数据集放哪
下载的test2k数据集应该放在什么目录下面
放哪都行,只要修改一下test的yaml里面的路径
cuda版本有要求么,我发现我在python里可以找到gpu,但在代码运行的时候显示没有cuda
只要能正常运行pytorch1.5+就可以
cuda版本有要求么,我发现我在python里可以找到gpu,但在代码运行的时候显示没有cuda
codes\options\options.py os.environ['CUDA_VISIBLE_DEVICES'] = gpu_list改成os.environ["CUDA_VISIBLE_DEVICES"] = "0" 另外在运行python test_ClassSR.py -opt options/test/test_ClassSR_SRResNet.yml的时候要对应把test_ClassSR.py执行的部分定义成函数(比如main),然后再放到 if name=='main': main() 执行,可能是因为windows多进程的问题...
cuda版本有要求么,我发现我在python里可以找到gpu,但在代码运行的时候显示没有cuda
codes\options\options.py os.environ['CUDA_VISIBLE_DEVICES'] = gpu_list改成os.environ["CUDA_VISIBLE_DEVICES"] = "0" 另外在运行python test_ClassSR.py -opt options/test/test_ClassSR_SRResNet.yml的时候要对应把test_ClassSR.py执行的部分定义成函数(比如main),然后再放到 if name=='main': main() 执行,可能是因为windows多进程的问题...
把cuda设备数设为0确实可以这个问题,但是这样会用cpu进行训练而不是GPU
我训练时遇到一些问题,在我开始训练后,一直卡在这个地方不训练,请问是你们有遇到这个问题吗?
@laonafahaodange @zhl98
Did you solve it? I also encountered the same issue. @msx-123
我训练时遇到一些问题,在我开始训练后,一直卡在这个地方不训练,请问是你们有遇到这个问题吗?
@laonafahaodange @zhl98
不好意思我还没有跑过训练的部分,可以看下GPU是否有占用?或者调试看看