update utils

layumi · Jan 24, 2024 · e9d37e1 · e9d37e1
1 parent 34158a0
commit e9d37e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils.py b/utils.py
@@ -1,4 +1,5 @@
 import torch.nn as nn
+import os
 from torch.nn.utils import fuse_conv_bn_eval
 
 class CrossEntropyLabelSmooth(nn.Module):