train cape_m encoders

2023-02-24 18:58:34 +00:00
parent 4713a1fa89
commit 0ce28a2e18
5 changed files with 107 additions and 3 deletions
@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_ae_body_cape_f import config
+from configs.iccv.config_ae_body_cape_m import config
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']

 import tensorflow as tf