feat: 更新README和训练脚本,简化环境设置并移除不必要的代码

This commit is contained in:
2025-01-13 18:26:38 +08:00
parent 3b606da46f
commit 7d70d85c60
2 changed files with 2 additions and 20 deletions
-1
View File
@@ -89,7 +89,6 @@ if __name__ == "__main__":
accelerator.wait_for_everyone()
model.eval()
accelerator = trainer.accelerator
from torch.utils.data import DataLoader