creating resnet101

auto_encoder_static.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.review.config_ae_static import config
+from configs.iccv.config_ae_static import config
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 
 import tensorflow as tf
@@ -55,7 +55,7 @@ uv2mesh = UV2Mesh(res=256, is_skirt=config['is_skirt'], is_smpl_orig=config['is_
 
 
 
-@tf.function
+# @tf.function
 def train_step(batch):
     with tf.GradientTape() as tape:
         pred = model([batch['uv_static'], batch['uv_mask']], training=True)
@@ -189,8 +189,7 @@ def test():
 
 
 if __name__ == '__main__':
-    init_wandb()
-    # allow_memory_growth()
+    # init_wandb()
     parser = argparse.ArgumentParser(description='')
     parser.add_argument('--name', type=str)
     args = vars(parser.parse_args())