From 19f7e0dfb989b6ab83159d1024a7520b85719d7c Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Thu, 26 Jan 2023 12:17:56 +0000 Subject: [PATCH] config update --- aaa_helpers.txt | 2 ++ configs/config_abl_pyr_32_skirt.py | 3 ++- cvae_pyramid.py | 4 ++-- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/aaa_helpers.txt b/aaa_helpers.txt index f8ea6ae..46242c6 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -1,3 +1,5 @@ +nohup python cvae_pyramid.py --name="lhunor_abl_pyr_32_no_curv_r1" > nohup/review/abl_pyr_32_no_curv_r1.out 2>&1 & + one final convolutional layer for smoothing, on the final coordinates skirts!!!! diff --git a/configs/config_abl_pyr_32_skirt.py b/configs/config_abl_pyr_32_skirt.py index 3e129a9..a250033 100644 --- a/configs/config_abl_pyr_32_skirt.py +++ b/configs/config_abl_pyr_32_skirt.py @@ -2,7 +2,8 @@ config = dict() config['gpu'] = '6' config['name'] = 'pyr_abl_32_skirt_no_curv_r1' -config['notes'] = 'review_1' +config['notes'] = 'no curv, 10p data' +config['tags'] = ['review_1'] # DON'T FORGET TO RESET SSIM LOSS USAGE # DON'T FORGET TO CHECK MODEL LOADING diff --git a/cvae_pyramid.py b/cvae_pyramid.py index af88b98..196944e 100644 --- a/cvae_pyramid.py +++ b/cvae_pyramid.py @@ -210,7 +210,7 @@ def plot_3d(mesh_gt, mesh_pred_unposed_levels, faces, mesh_posed, collision, col return fig -# @tf.function +@tf.function def train_step(batch): levels_input, conds = group_inputs_for_levels(batch, config) @@ -733,7 +733,7 @@ def speed_step(levels_input, conds): if __name__ == '__main__': - # init_wandb() + init_wandb() parser = argparse.ArgumentParser(description='') parser.add_argument('--name', type=str, default='lhunor') args = vars(parser.parse_args())