tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-15 04:13:57 +08:00
Code Activity

nerfacto-huge (#2003)

Browse Source 
An even larger version of nerfacto with scaled hashgrid resolution, MLP sizes, batch sizes, appearance embeddings, proposal sampler resolution, and more.
Also includes some minor tweaks to nerfacto-big configs
This commit is contained in:
Justin Kerr authored and GitHub committed 2023-06-15 11:05:53 -07:00
1 parent 26fc1c95dc
commit 8fad92c39a
5 files changed
+77 -41
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

nerfstudio/engine/trainer.py
+2 -4
View File
@@ -453,10 +453,8 @@ class Trainer:
"""
self.optimizers.zero_grad_all()
device_type: str = self.device.split(":")[0] if "cuda" in self.device else "cpu"
with torch.autocast(device_type=device_type, enabled=self.mixed_precision):
cpu_or_cuda_str: str = self.device.split(":")[0]
with torch.autocast(device_type=cpu_or_cuda_str, enabled=self.mixed_precision):
_, loss_dict, metrics_dict = self.pipeline.get_train_loss_dict(step=step)
loss = functools.reduce(torch.add, loss_dict.values())
self.grad_scaler.scale(loss).backward() # type: ignore