Skip to content

Commit e040735

Browse files
committed
Merge: [PyT/EffDet] Replace _six.string_classes with Actual Values
2 parents 0e2e315 + 0953478 commit e040735

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

PyTorch/Detection/Efficientdet/data/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ class CocoDetection(data.Dataset):
4343

4444
def __init__(self, root, ann_file, config, transform=None):
4545
super(CocoDetection, self).__init__()
46-
if isinstance(root, torch._six.string_classes):
46+
if isinstance(root, (str, bytes)):
4747
root = os.path.expanduser(root)
4848
self.root = root
4949
self.transform = transform

0 commit comments

Comments
 (0)