Colmap type of dataset (from "friends" project) (#56)

* code to process the friends data. need to add masking

* code to process the friends data. need to add masking

* black

* trying to fix black

* black fixed

* updating black version

scripts/run_train.py

@@ -105,7 +105,7 @@ def _distributed_worker(
     return output
 
 
-def _train(local_rank: int, world_size: int, config: DictConfig, cpu: bool = False) -> Any:
+def _train(local_rank: int, world_size: int, config: DictConfig) -> Any:
     """Main training function that sets up and runs the trainer per process
 
     Args:
@@ -116,7 +116,7 @@ def _train(local_rank: int, world_size: int, config: DictConfig, cpu: bool = Fal
     Returns:
         Any: TODO(): determine the return type
     """
-    trainer = Trainer(config, local_rank, world_size, cpu=cpu)
+    trainer = Trainer(config, local_rank, world_size)
     trainer.setup()
     trainer.train()
     return 0
@@ -145,7 +145,7 @@ def launch(
     world_size = num_machines * num_gpus_per_machine
     if world_size == 0:
         # Using only CPU and one process.
-        main_func(local_rank=0, world_size=0, config=config, cpu=True)
+        main_func(local_rank=0, world_size=0, config=config)
     elif world_size == 1:
         # Using one gpu and one process.
         try: