This commit is contained in:
2023-03-04 18:35:35 +00:00
parent b95ccc3ca1
commit 76cb457a54
7 changed files with 580 additions and 11 deletions
+3 -3
View File
@@ -1,5 +1,5 @@
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
os.environ["CUDA_VISIBLE_DEVICES"] = "3"
import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
@@ -36,10 +36,10 @@ from models.upscaler import UpscalerBasicConv as Upscaler
# !!!!!!!!!! CHANGE SKIRT OR NOT
low_res = 128
low_res = 256
is_skirt = False
is_cape = True
is_male = False
is_male = True
skirt_modifier = '_skirt' if is_skirt else ''
cape_modifier = '_cape' if is_cape else ''
gender_modifier = '' if not is_cape else ('_m' if is_male else '_f')