jchsiao2025 commited on
Commit
d1b0194
·
verified ·
1 Parent(s): 6be67e1

Update macbert/infer_all.py

Browse files
Files changed (1) hide show
  1. macbert/infer_all.py +2 -2
macbert/infer_all.py CHANGED
@@ -90,8 +90,8 @@ def test_epoch(travel_model, name_model, epoch, dataloader, tokenizer):
90
  def inference():
91
  travel_checkpoint_file = 'checkpoints/saved_checkpoints/travel_checkpoint15_train8000.pth.tar'
92
  name_checkpoint_file = 'checkpoints/saved_checkpoints/name_checkpoint17_train9000.pth.tar'
93
- ann_file_test = r'dataset/datagame_sms_stage1(in).csv'
94
- output_file = r'/both_macbertBase_20250731_2.csv'
95
  cache_dir = 'cache'
96
 
97
  model_cfg = {
 
90
  def inference():
91
  travel_checkpoint_file = 'checkpoints/saved_checkpoints/travel_checkpoint15_train8000.pth.tar'
92
  name_checkpoint_file = 'checkpoints/saved_checkpoints/name_checkpoint17_train9000.pth.tar'
93
+ ann_file_test = 'dataset/datagame_sms_stage1(in).csv'
94
+ output_file = 'both_macbertBase_20250731_2.csv'
95
  cache_dir = 'cache'
96
 
97
  model_cfg = {