Skip to content

Commit 0953478

Browse files
suiyoubinv-kkudrynski
authored andcommitted
[PyT/EffDet] Replace _six.string_classes with Actual Values
1 parent 441cf4b commit 0953478

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)