From 5839e18a72519222108e27d93411901f43c74847 Mon Sep 17 00:00:00 2001
From: huihun <781165206@qq.com>
Date: Tue, 26 Mar 2024 14:19:07 +0800
Subject: [PATCH] first commit

---
 src/trainer/pretrain.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/trainer/pretrain.py b/src/trainer/pretrain.py
index 63149fe..1f3d31f 100644
--- a/src/trainer/pretrain.py
+++ b/src/trainer/pretrain.py
@@ -3,8 +3,8 @@ import torch.nn as nn
 from torch.optim import Adam
 from torch.utils.data import DataLoader
 
-from model.bert import BERT
-from model.language_model import BERTLM
+from ..model.bert import BERT
+from ..model.language_model import BERTLM
 from .optim_schedule import ScheduledOptim
 
 import tqdm