remove things

Author: humanpose1

torch_points3d/.#trainer.py

@@ -1,7 +1,7 @@
 import hydra
 from hydra.core.global_hydra import GlobalHydra
 from omegaconf import OmegaConf
-from torch_points3d.trainer import Trainer
+from torch_points3d.trainer import LitTrainer
 
 OmegaConf.register_new_resolver("get_filename", lambda x: x.split('/')[-1])
 @hydra.main(config_path="conf", config_name="config")
@@ -10,7 +10,7 @@ def main(cfg):
     if cfg.pretty_print:
         print(OmegaConf.to_yaml(cfg))
 
-    trainer = Trainer(cfg)
+    trainer = LitTrainer(cfg)
     trainer.train()
 
 if __name__ == "__main__":