Update inference/style_transfer.py
Browse files
inference/style_transfer.py
CHANGED
@@ -301,7 +301,7 @@ class Mixing_Style_Transfer_Inference:
|
|
301 |
return whole_batch_data
|
302 |
|
303 |
|
304 |
-
def set_up()
|
305 |
os.environ['MASTER_ADDR'] = '127.0.0.1'
|
306 |
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
|
307 |
os.environ['MASTER_PORT'] = '8888'
|
|
|
301 |
return whole_batch_data
|
302 |
|
303 |
|
304 |
+
def set_up():
|
305 |
os.environ['MASTER_ADDR'] = '127.0.0.1'
|
306 |
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
|
307 |
os.environ['MASTER_PORT'] = '8888'
|