From 9512f9fe67ab9e8989bc8410c0eb88a937e0d2cb Mon Sep 17 00:00:00 2001 From: Zhen Liu Date: Wed, 2 Aug 2023 02:23:11 +0200 Subject: [PATCH] Update trainer.py --- lib/diffusion/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/diffusion/trainer.py b/lib/diffusion/trainer.py index 9dc5c93..5e6aec1 100644 --- a/lib/diffusion/trainer.py +++ b/lib/diffusion/trainer.py @@ -66,7 +66,7 @@ def train(config): print("sdf normalized or not: ", config.data.normalize_sdf) train_dataset = ShapeNetDMTetDataset(json_path, deform_scale=config.model.deform_scale, aug=True, grid_mask=mask, - filter_meta_path=config.data.filter_meta_path, normalize_sdf=config.data.normalize_sdf) + filter_meta_path=config.data.filter_meta_path, normalize_sdf=config.data.normalize_sdf, extension=config.data.extension) train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=config.training.batch_size,