search

nanoMFG / unet-sem

image segmentation for SVD graphene SEM images
MIT License
4 stars 1 forks source link

crash on ngpu > 1 #3

Closed dadamsncsa closed 5 years ago

dadamsncsa commented 5 years ago 
Traceback (most recent call last):
  File "main.py", line 31, in <module>
    run = TrainUNET(**kwargs)
  File "/home/dadams2/unet-sem/model.py", line 94, in __init__
    if self.ngpu > 1:
TypeError: unorderable types: list() > int()
dadamsncsa commented 5 years ago

Getting list type from kwargs.