work on torch==1.10.0

Hiroshiba · Nov 1, 2021 · b4b3d64 · b4b3d64
1 parent 893356c
commit b4b3d64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytorch_trainer/dataset/convert.py b/pytorch_trainer/dataset/convert.py
@@ -3,7 +3,7 @@
 import numpy
 import six
 import torch
-from torch._six import container_abcs
+from collections import abc as container_abcs
 
 import pytorch_trainer
 

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='pytorch-trainer',
-    version='1.3.1',
+    version='1.3.3',
     packages=find_packages(),
     url='https://github.com/Hiroshiba/pytorch-trainer',
     author='Kazuyuki Hiroshiba',