search

crowsonkb / style_transfer

Data-parallel image stylization using Caffe.
MIT License
113 stars 14 forks source link

AttributeError: module 'caffe' has no attribute 'set_random_seed' #22

Open Infinality opened 5 years ago

Infinality commented 5 years ago

I successfully compiled NVCaffe, however I'm getting this error when I run style_transfer.py:

Run 190122_141358 started. Initializing vgg19.caffemodel.

Watch the progress at: http://127.0.0.1:8000/

Starting 1 worker process(es). ======== Running on http://0.0.0.0:8000 ======== (Press CTRL+C to quit)

Scale 1, image size 256x256.

Preprocessing the style image(s)... Process ForkProcess-1: Traceback (most recent call last): File ".../style_transfer/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap self.run() File ".../style_transfer/lib/python3.6/multiprocessing/process.py", line 93, in run self._target(*self._args, **self._kwargs) File "style_transfer.py", line 185, in run caffe.set_random_seed(0) AttributeError: module 'caffe' has no attribute 'set_random_seed'