HXH-resonance

在运行python main.py train时出现如下问题,系统环境为ubuntu16.04+cuda9.0+cudnn7.0.5,百度之后发现该问题可能是因为cuda计算能力不够,cudnn需要计算能力达到3.0的cuda,但是cuda9.0的计算能力为2.1,是不足以支持的,但是在配置环境的时候网上有很多教程都是ubuntu16.04+cuda9.0+cudnn7.0.5,想问一下真的是cuda计算能力的问题吗还是别的问题

resnet model

1

can you share the resnet model to train [email protected]

请问 extract = { 'vgg': {'b':[21,33],'e':[1,3,5,7]}, #vgg -14 'resnet': {'b':[10,16,19],'e':[0,1,2]} } 这里resnet的参数是如何定下来的呢？如果改变提取特征的层除了需要更改这里的参数，还有别的参数需要更改吗？