working auto encoders

auto_encoder_static.py

@@ -83,7 +83,7 @@ def valid_step(batch):
 
 
 def plot_everything(batch: dict, type: str):
-    pred = model([batch['uv_static'], batch['uv_mask']])
+    pred = model([batch['uv_static'], batch['uv_mask']], training=False)
     
     mesh_pred = uv2mesh(pred)
     mesh_orig = batch['mesh_static']
@@ -151,7 +151,7 @@ def test():
 
 
 if __name__ == '__main__':
-    # init_wandb()
+    init_wandb()
     allow_memory_growth()
     parser = argparse.ArgumentParser(description='')
     parser.add_argument('--name', type=str)
@@ -178,5 +178,5 @@ if __name__ == '__main__':
     ckpt_dec = tf.train.Checkpoint(model=model.decoder)
     manager_dec = tf.train.CheckpointManager(ckpt_dec, 'checkpoints/' + config['name'] + '_dec', max_to_keep=3)
 
-    test()
-    # train()
+    # test()
+    train()