Update utils.py ——del torch._six

4c43f612 · dcuai · bc29c9e2 · 4c43f612
Commit 4c43f612 authored Nov 07, 2024 by dcuai
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

utils.py utils.py +1 -1

No files found.
--- a/utils.py
+++ b/utils.py
@@ -8,7 +8,7 @@
 import os
 import torch
 import torch.distributed as dist
-from torch._six import inf
+from torch import inf


 def load_checkpoint(config, model, optimizer, lr_scheduler, loss_scaler, logger):