laughtervv/SGPN

error when running valid.py

amiltonwong opened this issue · 1 comments

amiltonwong commented

Hi, @laughtervv ,

I got the following error when I run valid.py

(py2.7_tf1.4) root@milton-ThinkCentre-M93p:/data/code6/SGPN# python valid.py 
Traceback (most recent call last):
  File "valid.py", line 6, in <module>
    from utils.test_utils import *
  File "/data/code6/SGPN/utils/test_utils.py", line 310, in <module>
    magma_cmap = mpl.cm.get_cmap('magma')
AttributeError: 'module' object has no attribute 'cm'

I can run python train.py successfully before running valid.py.
Could you suggest me how to fix it?

THX!

mr-scrawley commented

I encountered the same problem and was able to fix it with an additional, explicit import:
import matplotlib.cm

  • 👍2