Spaces:

pkiage
/

fast_arbitrary_image_style_transfer

Runtime error

fast_arbitrary_image_style_transfer / src /model /utils.py

First commit

732b57d over 4 years ago

290 Bytes

import os


def use_low_resource_settings(option):
    if option == 'Yes':
        os.environ['CUDA_VISIBLE_DEVICES'] = ''
        os.environ['TFHUB_MODEL_LOAD_FORMAT'] = 'COMPRESSED'


def suppress_warnings(option):
    if option == 'Yes':
        os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

	import os


	def use_low_resource_settings(option):
	if option == 'Yes':
	os.environ['CUDA_VISIBLE_DEVICES'] = ''
	os.environ['TFHUB_MODEL_LOAD_FORMAT'] = 'COMPRESSED'


	def suppress_warnings(option):
	if option == 'Yes':
	os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'