瀏覽代碼

[PyT/EffDet] Replace _six.string_classes with Actual Values

Ao Tang 2 年之前
父節點
當前提交
0953478cec
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      PyTorch/Detection/Efficientdet/data/dataset.py

+ 1 - 1
PyTorch/Detection/Efficientdet/data/dataset.py

@@ -43,7 +43,7 @@ class CocoDetection(data.Dataset):
 
     def __init__(self, root, ann_file, config, transform=None):
         super(CocoDetection, self).__init__()
-        if isinstance(root, torch._six.string_classes):
+        if isinstance(root, (str, bytes)):
             root = os.path.expanduser(root)
         self.root = root
         self.transform = transform