From 817c748e8c44ed8c12c3e95de355610c094fade1 Mon Sep 17 00:00:00 2001 From: Xinchen Liu Date: Wed, 5 Apr 2023 09:07:49 +0800 Subject: [PATCH] Update build.py (#692) --- fastreid/data/build.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/fastreid/data/build.py b/fastreid/data/build.py index 7d73820e..23ac8d62 100644 --- a/fastreid/data/build.py +++ b/fastreid/data/build.py @@ -8,7 +8,14 @@ import os import torch -from torch._six import string_classes +TORCH_MAJOR = int(torch.__version__.split('.')[0]) +TORCH_MINOR = int(torch.__version__.split('.')[1]) + +if TORCH_MAJOR == 1 and TORCH_MINOR < 8: + from torch._six import string_classes +else: + string_classes = str + from collections import Mapping from fastreid.config import configurable