From c79a9a3be72e31260f64754c70a77a6f46fb4618 Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Fri, 27 Jan 2023 10:36:56 -0800 Subject: [PATCH] Lint --- nerfstudio/models/dreamfusion.py | 2 +- nerfstudio/pipelines/dreamfusion_pipeline.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/nerfstudio/models/dreamfusion.py b/nerfstudio/models/dreamfusion.py index b52c80ad4..136df8ffc 100644 --- a/nerfstudio/models/dreamfusion.py +++ b/nerfstudio/models/dreamfusion.py @@ -77,7 +77,7 @@ class DreamFusionModelConfig(ModelConfig): sphere_collider: bool = True """Use spherical collider instead of box""" target_transmittance_start: float = 0.8 - """target transmittance for opacity penalty. This is the percent of the scene that is + """target transmittance for opacity penalty. This is the percent of the scene that is background when rendered at the start of training""" target_transmittance_end: float = 0.4 """target transmittance for opacity penalty. This is the percent of the scene that is diff --git a/nerfstudio/pipelines/dreamfusion_pipeline.py b/nerfstudio/pipelines/dreamfusion_pipeline.py index 1bc12bef1..efbe14b65 100644 --- a/nerfstudio/pipelines/dreamfusion_pipeline.py +++ b/nerfstudio/pipelines/dreamfusion_pipeline.py @@ -137,7 +137,7 @@ class DreamfusionPipeline(VanillaPipeline): else: text_embedding = self.base_text_embedding - with torch.autocast(device_type='cuda', dtype=torch.float16): + with torch.autocast(device_type="cuda", dtype=torch.float16): sds_loss, latents, grad = self.sd.sds_loss( text_embedding.to(self.sd_device), train_output.to(self.sd_device),