From cc723222dc3f3d045c657318c8fba4e0a10e2b13 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 28 Jan 2023 19:57:23 +0000 Subject: [PATCH] use jumpsuit overfit model --- configs/config_abl_overfit_jumpsuit_2.py | 4 ++-- cvae.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/configs/config_abl_overfit_jumpsuit_2.py b/configs/config_abl_overfit_jumpsuit_2.py index 2125f42..86feac6 100644 --- a/configs/config_abl_overfit_jumpsuit_2.py +++ b/configs/config_abl_overfit_jumpsuit_2.py @@ -19,8 +19,8 @@ config['only_curvatures'] = True ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False) assert ablation_count == 1 or ablation_count == 0 -config['gpu'] = '5' -config['name'] = 'abl_overfit_jumpsuit_2' +config['gpu'] = '3' +config['name'] = 'abl_overfit_jumpsuit_2_r4' config['notes'] = 'remove other conditions from input as well' config['avg_garment_path'] = 'custom_data/train_jumpsuit_30k_no_shape_avg_uv.npy' #custom_data/train_no_shape_avg_uv.npy # train_no_shape_avg_uv train_no_shape_one_hot_avg_uv diff --git a/cvae.py b/cvae.py index 59d7c75..a2de40f 100644 --- a/cvae.py +++ b/cvae.py @@ -1,6 +1,6 @@ import os import sys -from configs.config_abl_all_post import config +from configs.config_abl_overfit_jumpsuit_2 import config os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] import tensorflow as tf @@ -41,7 +41,7 @@ from losses import collision_loss as collision_penalty from custom_utils.custom_utils import batch_to_dict, load_model, plot_3d, UV2Mesh, get_summary_html, maskFacesBatch, _maskFaces, plot_3d_collision from dataset import DatasetFactory, StaticDataset from custom_utils.smpl import SMPL -from models.cvae_model import CVAENew as CVAE +from models.cvae_model import CVAE def init_wandb():