viX-shaw

Results 2 comments of viX-shaw

When I ran train_ua.py, after following your instructions, I ran into CUDA memory error #22 . Can anybody help??

Model type detected: global pooling Model cut at layer: 1 --------------------------------------------------------------------------- ValueError Traceback (most recent call last) in () ----> 1 new_model = to_heatmap(model) 2 idx = 0 # The...