finding tiny faces中训练 hr_res101('train')出现问题

>> hr_res101('train')

ans =

    'models\widerface-resnet-101-simple-sample256-posfrac0.5-N25-bboxreg-cluster-scaled'

Trying to initialize the structure of resnet-101-simple
Unknown model: cannot initialize.
Loading pretrained weights from ./trained_models/imagenet-resnet-101-dag.mat
Loaded imdb from data\widerface\imdb.mat
cluster path: data\widerface\RefBox_N25_scaled.mat

opts = 

  包含以下字段的 struct:

      keepDilatedZeros: 0
             inputSize: [500 500]
          learningRate: [1×30 double]
               trainFn: '@cnn_train_dag_hardmine'
         batchGetterFn: '@cnn_get_batch_hardmine'
          freezeResNet: 0
                   tag: ''
            clusterNum: 25
           clusterName: 'scaled'
               bboxReg: 1
            skipLRMult: [0 1 0.1000]
            sampleSize: 256
           posFraction: 0.5000
             posThresh: 0.7000
             negThresh: 0.3000
                border: [0 0]
     pretrainModelPath: './trained_models/imagenet-resnet-101-dag.mat'
               dataDir: 'data\widerface'
             modelType: 'resnet-101-simple'
           networkType: 'dagnn'
    batchNormalization: 1
      weightInitMethod: 'gaussian'
        minClusterSize: [10 10]
        maxClusterSize: [Inf Inf]
                expDir: 'models\widerface-resnet-101-simple-sample256-posfrac0.5-N25-bboxreg-cluster-scaled'
             batchSize: 48
         numSubBatches: 1
             numEpochs: 50
                  gpus: [1 2 3 4]
       numFetchThreads: 8
                  lite: 0
              imdbPath: 'data\widerface\imdb.mat'
                 train: [1×1 struct]


ans = 

  包含以下字段的 struct:

                gpus: [1 2 3 4]
           batchSize: 48
       numSubBatches: 1
           numEpochs: 50
        learningRate: [1×30 double]
    keepDilatedZeros: 0

Start using dagnn.DetLoss for loss
cnn_train_dag_hardmine: resetting GPU
错误使用 cnn_train_dag_hardmine>prepareGPUs (line 616)
Error detected on workers 3 4.

出错 cnn_train_dag_hardmine (line 132)
  prepareGPUs(opts, epoch == start+1) ;

出错 cnn_widerface (line 212)
[net, info] = trainFn(net, imdb, getBatchFn(batchGetter, opts, net.meta), ...

出错 hr_res101 (line 41)
    cnn_widerface('inputSize', inputSize, ...

原因:
    错误使用 gpuDevice (line 26)
    Invalid CUDA device id: 3. Select a device id from the range 1:1.
    错误使用 gpuDevice (line 26)
    Invalid CUDA device id: 4. Select a device id from the range 1:1.
 

我电脑就一个gpu,这个gpuDevice怎么设置变成一个啊?
程序似乎是4个gpu什么的,没理解,有人知道怎么解决吗?
困扰了好久

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

百年孤独百年

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值