PROSAGA码农传奇-深度学习-Caffe

Caffe - 中止训练

作者: 石茶
发布时间: 2024-07-13 07:49:02 (22天前)
转自：

<div class =“post-text”itemprop =“text”>
  
    你正在使用
     <code>
 solver_mode: CPU
 </code>
    ，你在这次培训开始时检查CPU内存利用率了吗？
  
  
    你正在使用非常高的培训
     <code>
 batch_size
 </code>
  
   <pre>
 <code>
 layer {
 name: "training_train"
 type: "Data"
 top: "data"
 top: "label"
 include {
 phase: TRAIN
 }
 transform_param {
 mean_file: "datasets/mean_training_image.binaryproto"
 }
 data_param {
 source: "datasets/training_set_lmdb"
 batch_size: 400
 backend: LMDB
 }
 }

</code>
 </pre>
  
    所以看起来像这需要内存
     <code>
 batch_size: 400
 </code>
     在您的系统CPU上不可用。
  
  
    所以减少
     <code>
 batch_size
 </code>
     并提供培训。让我们说吧
     <code>
 batch_size: 20
 </code>
    。
  
  
    一旦了解了系统上的可用内存，就可以大致计算出来
     <code>
 batch_size
 </code>
     你可以使用。
  
</DIV>