默认是用cpu？ - Githubissues

BucherLi commented 5 years ago

运行bert-base-ner-train 训练结果，为啥没用到gpu，tf版本1.12， python版本3.6

       ARG   VALUE

 batch_size = 64
bert_config_file =~/chinese_L-12_H-768_A-12/bert_config.json
            cell = lstm
           clean = True
            clip = 0.5
        data_dir = ~/BERT-BiLSTM-CRF-NER-master/train
      device_map = 0
         do_eval = True
   do_lower_case = True
      do_predict = True
        do_train = True
    dropout_rate = 0.5
 filter_adam_var = False
 init_checkpoint =~/chinese_L-12_H-768_A-12/bert_model.ckpt
      label_list = None
   learning_rate = 1e-05
       lstm_size = 128
  max_seq_length = 128
             ner = ner
      num_layers = 1
num_train_epochs = 10
      output_dir = model

save_checkpoints_steps = 500 save_summary_steps = 500 verbose = False vocab_file =~/chinese_L-12_H-768_A-12/vocab.txt warmup_proportion = 0.1

WARNING:tensorflow:From /data/home/user00/python36.8/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py:263: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version. Instructions for updating: Colocations handled automatically by placer. WARNING:tensorflow:From /data/home/user00/python36.8/lib/python3.6/site-packages/bert_base/train/bert_lstm_ner.py:334: to_int32 (from tensorflow.python.ops.math_ops) is deprecated and will be removed in a future version. Instructions for updating: Use tf.cast instead. shape of input_ids (64, 128) WARNING:tensorflow:From /data/home/user00/python36.8/lib/python3.6/site-packages/bert_base/bert/modeling.py:359: calling dropout (from tensorflow.python.ops.nn_ops) with keep_prob is deprecated and will be removed in a future version. Instructions for updating: Please use rate instead of keep_prob. Rate should be set to rate = 1 - keep_prob. WARNING:tensorflow:From /data/home/user00/python36.8/lib/python3.6/site-packages/bert_base/bert/modeling.py:673: dense (from tensorflow.python.layers.core) is deprecated and will be removed in a future version. Instructions for updating: Use keras.layers.dense instead. WARNING:tensorflow:From /data/home/user00/python36.8/lib/python3.6/site-packages/tensorflow/contrib/crf/python/ops/crf.py:213: dynamic_rnn (from tensorflow.python.ops.rnn) is deprecated and will be removed in a future version. Instructions for updating: Please use keras.layers.RNN(cell), which is equivalent to this API WARNING:tensorflow:From /data/home/user00/python36.8/lib/python3.6/site-packages/tensorflow/python/training/learning_rate_decay_v2.py:321: div (from tensorflow.python.ops.math_ops) is deprecated and will be removed in a future version. Instructions for updating: Deprecated in favor of operator or tf.math.divide. 2019-06-25 21:44:04.525045: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA 2019-06-25 21:44:04.530741: I tensorflow/core/platform/profile_utils/cpu_utils.cc:94] CPU Frequency: 2199860000 Hz 2019-06-25 21:44:04.531249: I tensorflow/compiler/xla/service/service.cc:150] XLA service 0x9587ca0 executing computations on platform Host. Devices: 2019-06-25 21:44:04.531282: I tensorflow/compiler/xla/service/service.cc:158] StreamExecutor device (0): ,