This commit is contained in:
2023-05-23 10:35:14 +00:00
parent 1620f98d28
commit 19e868f271
3 changed files with 6 additions and 6 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
import os
import sys
from configs.iccv.config_abl_pyr_32_tshirt import config
from configs.iccv.config_abl_pyr_32_64_128_256_seq import config
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']