python train.py --network resnet50

parser.add_argument('--pretrained', dest='pretrained', help='pretrained model prefix',
                        default=os.path.join(os.getcwd(), 'model', 'resnet-50'), type=str)

 

 

使用预训练模型Inception_v3进行训练。

尝试一:batch-size=64,lr=0.005,--lr-steps='',--end-epoch=280

MXNet-SSD 再识--Inception_v3

发现其实很早的时候就已经趋于平衡了,所以可以尝试在80和160对学习率进行修改。

尝试二:batch-size=128,lr=0.005,--lr-steps='80,160',--end-epoch=260

 

报错:

src/storage/./pooled_storage_manager.h:143: cudaMalloc failed: out of memory

修改batch_size的大小。

相关文章:

  • 2021-09-02
  • 2021-04-11
  • 2021-06-04
  • 2021-09-27
  • 2021-05-30
  • 2021-10-19
  • 2021-10-12
  • 2021-12-03
猜你喜欢
  • 2021-11-23
  • 2021-09-17
  • 2022-02-01
  • 2021-06-03
  • 2022-12-23
  • 2021-12-16
  • 2021-08-31
相关资源
相似解决方案