add fix for torch 1.0 on RTD
from __future__ import absolute_import, division, print_function
+import os
+
import torch
+if 'READTHEDOCS' not in os.environ:
+ # RTD is running 0.4.1 due to a memory issue with pytorch 1.0
assert torch.__version__.startswith('1.')
