Update macbert/infer_all.py
Browse files- 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 =
|
| 94 |
-
output_file =
|
| 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 = {
|