-
Notifications
You must be signed in to change notification settings - Fork 84
Open
Description
I did it and got an error
import caffemodel2pytorch as caffe
caffe.set_mode_gpu()
caffe.set_device(0)
AttributeError Traceback (most recent call last)
in ()
1 get_ipython().magic('cd /content/caffemodel2pytorch')
2 import caffemodel2pytorch as caffe
----> 3 caffe.set_mode_gpu()
4 caffe.set_device(0)
AttributeError: module 'caffemodel2pytorch' has no attribute 'set_mode_gpu'
Metadata
Metadata
Assignees
Labels
No labels