From edc189fa969e9d4d1e9574c4bec71ddd2327c90d Mon Sep 17 00:00:00 2001
From: Zhikang Niu <73390819+ZhikangNiu@users.noreply.github.com>
Date: Sun, 13 Oct 2024 10:04:13 +0800
Subject: [PATCH] Update trainer.py

---
 model/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model/trainer.py b/model/trainer.py
index 4914f30..c0804fd 100644
--- a/model/trainer.py
+++ b/model/trainer.py
@@ -172,7 +172,7 @@ class Trainer:
             train_dataloader = DataLoader(train_dataset, collate_fn=collate_fn, num_workers=num_workers, pin_memory=True,
                                           batch_sampler=batch_sampler)
         else:
-            raise ValueError(f"batch_size_type must be either 'sample' or 'frame', but recieved {self.batch_size_type}")
+            raise ValueError(f"batch_size_type must be either 'sample' or 'frame', but received {self.batch_size_type}")
         
         #  accelerator.prepare() dispatches batches to devices;
         #  which means the length of dataloader calculated before, should consider the number of devices