Got error when import geffnet using Google Colab

[cwza]

When I import geffnet in Google Colab.
I got some error:

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-2-1dbdac5fd1e4> in <module>()
----> 1 import geffnet

3 frames
/usr/local/lib/python3.6/dist-packages/geffnet/layers.py in versiontuple(v)
      5 
      6 def versiontuple(v):
----> 7     return tuple(map(int, (v.split("."))))[:3]
      8 
      9 

ValueError: invalid literal for int() with base 10: '0+cu100'

I think it is because the torch.version string is '1.3.0+cu100' in Google Colab.
Maybe we can use some regular expression to parse this??

> /usr/local/lib/python3.6/dist-packages/geffnet/layers.py(7)versiontuple()
      5 
      6 def versiontuple(v):
----> 7     return tuple(map(int, (v.split("."))))[:3]
      8 
      9 

ipdb> v
'1.3.0+cu100'
ipdb> v.split(".")
['1', '3', '0+cu100']

[rwightman]

Thanks, that was a half-assed attempt that I was planning to remove. Now that I'm supporting more advanced torchscript usage probably no point in attempting to support anything below pytorch 1.2 anyhow. It's been removed.