mirror
/
MeshDiffusion
kopia lustrzana https://github.com/lzzcd001/MeshDiffusion
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

Update trainer.py

main
Zhen Liu 2023-08-02 02:23:11 +02:00 zatwierdzone przez GitHub
rodzic d571b74081
commit 9512f9fe67
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 1 dodań i 1 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

2
lib/diffusion/trainer.py
Wyświetl plik

@ -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,