diff --git a/bert_pytorch/trainer/pretrain.py b/bert_pytorch/trainer/pretrain.py index 0b882dd..670a389 100644 --- a/bert_pytorch/trainer/pretrain.py +++ b/bert_pytorch/trainer/pretrain.py @@ -75,7 +75,7 @@ def iteration(self, epoch, data_loader, train=True): """ loop over the data_loader for training or testing if on train status, backward operation is activated - and also auto save the model every peoch + and also auto save the model every epoch :param epoch: current epoch index :param data_loader: torch.utils.data.DataLoader for iteration