2 Ревизии 1b996c453c ... 2eebbe6c52

Автор SHA1 Съобщение Дата
  fangjiasheng 2eebbe6c52 Merge remote-tracking branch 'origin/master' преди 9 месеца
  fangjiasheng 8a20f53976 表头识别模型torch版更新 преди 9 месеца
променени са 2 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. BIN
      BiddingKG/dl/table_head/model_40_0.951.pth
  2. 1 1
      BiddingKG/dl/table_head/predict_torch.py

BIN
BiddingKG/dl/table_head/model_40_0.951.pth


+ 1 - 1
BiddingKG/dl/table_head/predict_torch.py

@@ -9,7 +9,7 @@ from BiddingKG.dl.table_head.models.model_torch import TableHeadModel
 from BiddingKG.dl.table_head.pre_process_torch import CustomDatasetTiny40, set_same_table_head, set_label
 
 device = torch.device("cpu")
-model_path = os.path.abspath(os.path.dirname(__file__)) + '/model_40_0.959.pth'
+model_path = os.path.abspath(os.path.dirname(__file__)) + '/model_40_0.951.pth'
 batch_size = 1