luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
luni
2022-09-13 09:23:01 +00:00
parent 977e49900d
commit 440444cd82
19 changed files with 1630 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cvae_curv.py
+2 -1
View File
@@ -35,7 +35,7 @@ from losses import collision_loss as collision_penalty
from dataset import DatasetFactory, StaticDataset
from custom_utils.custom_utils import batch_to_dict
from custom_utils.smpl import SMPL
from models.cvae_curv import CVAE
from models.cvae_curv_model import CVAE
name='cvae_curv_2_2_2'
@@ -48,6 +48,7 @@ fixed deoriented bodies and need retraining (ae_body_6 with 50k iteration), adde
higher starting kl coeff (0.1)')
artifact = wandb.Artifact('code', type='code')
artifact.add_file('cvae_curv.py')
artifact.add_file('models/cvae_curv_model.py')
artifact.add_file('dataset.py')
artifact.add_file('losses.py')
artifact.add_file('convnext/models/convnext_tf.py')