From 631a694f6bc6b324685bcdd1ab32d8d381e4f7c1 Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Thu, 21 Apr 2022 11:41:15 -0700 Subject: [PATCH] fix lint errors --- mattport/nerf/trainer.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/mattport/nerf/trainer.py b/mattport/nerf/trainer.py index 74e7d01f5..d1035a193 100644 --- a/mattport/nerf/trainer.py +++ b/mattport/nerf/trainer.py @@ -17,13 +17,12 @@ from mattport.nerf.dataset.image_dataset import ImageDataset, collate_batch from mattport.nerf.dataset.utils import get_dataset_inputs from mattport.utils.decorators import check_main_thread -logging.getLogger('PIL').setLevel(logging.WARNING) +logging.getLogger("PIL").setLevel(logging.WARNING) + class Trainer: """Training class""" - # pylint: disable=too-many-instance-attributes - def __init__(self, config: DictConfig, local_rank: int = 0, world_size: int = 1): self.config = config self.local_rank = local_rank @@ -131,4 +130,4 @@ class Trainer: graph_outputs = self.graph(ray_indices) batch.pixels = batch.pixels.to(f"cuda:{self.local_rank}") losses = self.graph.module.get_losses(batch, graph_outputs) - # logging.info(losses) + logging.info(losses)