Code cosmetics (#88)

* Update main.py

* Update main.py

* [skip ci] updated main

Author: vfdev-5

templates/gan/main.py

@@ -6,7 +6,6 @@
 from datetime import datetime
 from pathlib import Path
 from typing import Any
-from ignite.contrib.handlers.wandb_logger import WandBLogger
 
 import torch
 import ignite.distributed as idist
@@ -215,6 +214,8 @@ def create_plots(engine):
     # ------------------------------------------------------------
 
     if rank == 0:
+        from ignite.contrib.handlers.wandb_logger import WandBLogger
+
         if isinstance(logger_handler, WandBLogger):
             # why handle differently for wandb ?
             # See : https://github.com/pytorch/ignite/issues/1894

templates/image_classification/main.py

@@ -5,7 +5,6 @@
 from datetime import datetime
 from pathlib import Path
 from typing import Any
-from ignite.contrib.handlers.wandb_logger import WandBLogger
 
 import ignite.distributed as idist
 from ignite.engine.events import Events
@@ -206,6 +205,8 @@ def _():
     # ------------------------------------------------------------
 
     if rank == 0:
+        from ignite.contrib.handlers.wandb_logger import WandBLogger
+
         if isinstance(logger_handler, WandBLogger):
             # why handle differently for wandb ?
             # See : https://github.com/pytorch/ignite/issues/1894

templates/single/main.py

@@ -5,7 +5,6 @@
 from datetime import datetime
 from pathlib import Path
 from typing import Any
-from ignite.contrib.handlers.wandb_logger import WandBLogger
 
 import ignite.distributed as idist
 from ignite.engine.events import Events
@@ -185,6 +184,8 @@ def _():
     # ------------------------------------------------------------
 
     if rank == 0:
+        from ignite.contrib.handlers.wandb_logger import WandBLogger
+
         if isinstance(logger_handler, WandBLogger):
             # why handle differently for wandb ?
             # See : https://github.com/pytorch/ignite/issues/1894