File tree Expand file tree Collapse file tree 10 files changed +12
-47
lines changed
template-text-classification
template-vision-classification
template-vision-segmentation Expand file tree Collapse file tree 10 files changed +12
-47
lines changed Original file line number Diff line number Diff line change 44
55#::: if (it.logger) { :::#
66if rank == 0 :
7- from ignite .contrib .handlers .wandb_logger import WandBLogger
8-
9- if isinstance (exp_logger , WandBLogger ):
10- # why handle differently for wandb?
11- # See: https://github.com/pytorch/ignite/issues/1894
12- exp_logger .finish ()
13- elif exp_logger :
14- exp_logger .close ()
7+ exp_logger .close ()
158#::: } :::#
169
1710#::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 11torch >= 1.8.0
22torchvision >= 0.9.0
3- pytorch-ignite >= 0.4.4
3+ pytorch-ignite >= 0.4.5
44pyyaml
55
66#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#
Original file line number Diff line number Diff line change 55import ignite .distributed as idist
66import yaml
77from data import setup_data
8- from ignite .contrib .handlers import LRScheduler , PiecewiseLinear
98from ignite .engine import Events
9+ from ignite .handlers import LRScheduler , PiecewiseLinear
1010from ignite .metrics import Accuracy , Loss
1111from ignite .utils import manual_seed
1212from models import TransformerModel
@@ -157,14 +157,7 @@ def _():
157157 #::: if (it.logger) { :::#
158158 # close logger
159159 if rank == 0 :
160- from ignite .contrib .handlers .wandb_logger import WandBLogger
161-
162- if isinstance (exp_logger , WandBLogger ):
163- # why handle differently for wandb?
164- # See: https://github.com/pytorch/ignite/issues/1894
165- exp_logger .finish ()
166- elif exp_logger :
167- exp_logger .close ()
160+ exp_logger .close ()
168161 #::: } :::#
169162 #
170163 #::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 11torch >= 1.8.0
22torchvision >= 0.9.0
3- pytorch-ignite >= 0.4.4
3+ pytorch-ignite >= 0.4.5
44pyyaml
55transformers
66datasets
Original file line number Diff line number Diff line change @@ -113,14 +113,7 @@ def _():
113113 #::: if (it.logger) { :::#
114114 # close logger
115115 if rank == 0 :
116- from ignite .contrib .handlers .wandb_logger import WandBLogger
117-
118- if isinstance (exp_logger , WandBLogger ):
119- # why handle differently for wandb?
120- # See: https://github.com/pytorch/ignite/issues/1894
121- exp_logger .finish ()
122- elif exp_logger :
123- exp_logger .close ()
116+ exp_logger .close ()
124117 #::: } :::#
125118 #
126119 #::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 11torch >= 1.8.0
22torchvision >= 0.9.0
3- pytorch-ignite >= 0.4.4
3+ pytorch-ignite >= 0.4.5
44pyyaml
55
66#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#
Original file line number Diff line number Diff line change @@ -167,14 +167,7 @@ def _():
167167 #::: if (it.logger) { :::#
168168 # close logger
169169 if rank == 0 :
170- from ignite .contrib .handlers .wandb_logger import WandBLogger
171-
172- if isinstance (exp_logger , WandBLogger ):
173- # why handle differently for wandb?
174- # See: https://github.com/pytorch/ignite/issues/1894
175- exp_logger .finish ()
176- elif exp_logger :
177- exp_logger .close ()
170+ exp_logger .close ()
178171 #::: } :::#
179172 #
180173 #::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 11torch >= 1.8.0
22torchvision >= 0.9.0
3- pytorch-ignite >= 0.4.4
3+ pytorch-ignite >= 0.4.5
44pyyaml
55
66#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#
Original file line number Diff line number Diff line change 55import ignite .distributed as idist
66import yaml
77from data import denormalize , setup_data
8- from ignite .contrib .handlers import LRScheduler
98from ignite .engine import Events
9+ from ignite .handlers import LRScheduler
1010from ignite .metrics import ConfusionMatrix , IoU , mIoU
1111from ignite .utils import manual_seed
1212from models import setup_model
@@ -176,14 +176,7 @@ def _():
176176 #::: if (it.logger) { :::#
177177 # close logger
178178 if rank == 0 :
179- from ignite .contrib .handlers .wandb_logger import WandBLogger
180-
181- if isinstance (exp_logger , WandBLogger ):
182- # why handle differently for wandb?
183- # See: https://github.com/pytorch/ignite/issues/1894
184- exp_logger .finish ()
185- elif exp_logger :
186- exp_logger .close ()
179+ exp_logger .close ()
187180 #::: } :::#
188181 #
189182 #::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 11torch >= 1.8.0
22torchvision >= 0.9.0
3- pytorch-ignite >= 0.4.4
3+ pytorch-ignite >= 0.4.5
44pyyaml
55albumentations
66image_dataset_viz
You can’t perform that action at this time.
0 commit comments