Update 2 files

- /trainer.py
- /trainer.cli.py

trainer.cli.py

@@ -4,6 +4,8 @@ from argparse import ArgumentParser
 from util import ConfigParser
 from logger import Wandb
 
+from trainer import Trainer
+
 
 
 parser = ArgumentParser(
@@ -20,6 +22,6 @@ if __name__ == '__main__':
 
 
     config = ConfigParser(args.config_path).config
-    wb_config = config.wandb
+    
 
-    wandb = Wandb(wb_config)
+    trainer = Trainer(config)

trainer.py

@@ -1,4 +1,9 @@
+from logger import Wandb
+
 
 class Trainer:
     def __init__(self, config: Config):
-        pass
+        self.__dict__ = dict(config.__dict__)
+
+        self.wandb = Wandb(config.wandb)
+