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)