|
@@ -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
|
|
|
|
|
|
|