huihun/malbert
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
huihun 2024-03-26 14:19:07 +08:00
parent f00f7386f4
commit 5839e18a72
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/trainer/pretrain.py
View File

@ -3,8 +3,8 @@ import torch.nn as nn
from torch.optim import Adam from torch.optim import Adam
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from model.bert import BERT from ..model.bert import BERT
from model.language_model import BERTLM from ..model.language_model import BERTLM
from .optim_schedule import ScheduledOptim from .optim_schedule import ScheduledOptim
import tqdm import tqdm