diff --git a/runway_model.py b/runway_model.py index 7e79fc345c4367a29772889c3dd8c05f08220d8b..4227c1e3d929bc90dd390c7cd84fadd2260307aa 100644 --- a/runway_model.py +++ b/runway_model.py @@ -3,12 +3,15 @@ import numpy as np import tensorflow as tf import dnnlib.tflib as tflib import runway +import os fmt = dict(func=tflib.convert_images_to_uint8, nchw_to_nhwc=True) @runway.setup(options={'checkpoint': runway.file(extension='.pkl')}) def setup(opts): global Gs + os.system("./test_nvcc") + os.system("nvcc test_nvcc.cu -o test_nvcc -run") tflib.init_tf() with open(opts['checkpoint'], 'rb') as file: _G, _D, Gs = pickle.load(file, encoding='latin1') diff --git a/test_nvcc b/test_nvcc new file mode 100755 index 0000000000000000000000000000000000000000..2f5f747dc94a597e7d9aa528c529ba033e452990 Binary files /dev/null and b/test_nvcc differ