Fix PT 1.8 compatibility issue

.gitignore

@@ -2,6 +2,7 @@
 __pycache__/
 *.py[cod]
 *$py.class
+data/
 
 # C extensions
 *.so
@@ -163,3 +164,8 @@ tests/test-reports/
 
 # MLflow
 mlruns/
+
+
+# Downloaded Data through Scripts
+VOCdevkit/
+*.tar

pl_bolts/datamodules/async_dataloader.py

@@ -1,10 +1,11 @@
+import collections.abc as container_abcs
 import re
 from queue import Queue
 from threading import Thread
 from typing import Any, Optional, Union
 
 import torch
-from torch._six import container_abcs, string_classes
+from torch._six import string_classes
 from torch.utils.data import DataLoader, Dataset
 
 

pl_bolts/datasets/imagenet_dataset.py

@@ -2,14 +2,14 @@
 import hashlib
 import os
 import shutil
+import sys
 import tarfile
 import tempfile
 import zipfile
 from contextlib import contextmanager
 
 import numpy as np
 import torch
-from torch._six import PY3
 
 from pl_bolts.utils import _TORCHVISION_AVAILABLE
 from pl_bolts.utils.warnings import warn_missing_pkg
@@ -246,6 +246,8 @@ def extract_archive(from_path, to_path=None, remove_finished=False):
     if to_path is None:
         to_path = os.path.dirname(from_path)
 
+    PY3 = sys.version_info[0] == 3
+
     if _is_tar(from_path):
         with tarfile.open(from_path, 'r') as tar:
             tar.extractall(path=to_path)

requirements.txt

@@ -1,2 +1,2 @@
-torch>=1.6
+torch==1.8
 pytorch-lightning>=1.1.1, <1.2

requirements/models.txt

@@ -1,4 +1,4 @@
-torchvision>=0.7
+torchvision==0.9.0
 scikit-learn>=0.23
 Pillow
 opencv-python