This commit is contained in:
2023-03-04 20:32:16 +00:00
parent 76cb457a54
commit dfdf8d408e
4 changed files with 17 additions and 16 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
import os
import sys
from configs.iccv.config_abl_pyr_32_64_128_skin import config
from configs.iccv.config_abl_pyr_32_64_128_reg import config
os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
import tensorflow as tf