This commit is contained in:
2023-03-15 17:53:57 +00:00
parent 30e33bb037
commit 7ee10aa7a2
+1 -1
View File
@@ -1,6 +1,6 @@
import os import os
import sys import sys
from configs.iccv.config_abl_pyr_32_64_128_256_seq import config from configs.iccv.config_abl_pyr_32_64_128_seq import config
os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
import tensorflow as tf import tensorflow as tf