replace backbone with resnet18

This commit is contained in:
2023-01-25 16:49:01 +00:00
parent ffdc60062d
commit 2305a0ee23
4 changed files with 303 additions and 3 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
import os
import sys
from configs.config_ae_static import config
from configs.review.config_ae_static import config
os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
import tensorflow as tf