Skip to content

Commit 22dd969

Browse files
authored
chore(deps): bump pytorch-ignite to 0.4.5 (#163)
1 parent 1564bb9 commit 22dd969

File tree

10 files changed

+12
-47
lines changed

10 files changed

+12
-47
lines changed

src/templates/template-common/main.py

+1-8
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,7 @@
44

55
#::: if (it.logger) { :::#
66
if 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) { :::#

src/templates/template-common/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
torch>=1.8.0
22
torchvision>=0.9.0
3-
pytorch-ignite>=0.4.4
3+
pytorch-ignite>=0.4.5
44
pyyaml
55

66
#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#

src/templates/template-text-classification/main.py

+2-9
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
import ignite.distributed as idist
66
import yaml
77
from data import setup_data
8-
from ignite.contrib.handlers import LRScheduler, PiecewiseLinear
98
from ignite.engine import Events
9+
from ignite.handlers import LRScheduler, PiecewiseLinear
1010
from ignite.metrics import Accuracy, Loss
1111
from ignite.utils import manual_seed
1212
from 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) { :::#

src/templates/template-text-classification/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
torch>=1.8.0
22
torchvision>=0.9.0
3-
pytorch-ignite>=0.4.4
3+
pytorch-ignite>=0.4.5
44
pyyaml
55
transformers
66
datasets

src/templates/template-vision-classification/main.py

+1-8
Original file line numberDiff line numberDiff 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) { :::#

src/templates/template-vision-classification/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
torch>=1.8.0
22
torchvision>=0.9.0
3-
pytorch-ignite>=0.4.4
3+
pytorch-ignite>=0.4.5
44
pyyaml
55

66
#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#

src/templates/template-vision-dcgan/main.py

+1-8
Original file line numberDiff line numberDiff 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) { :::#

src/templates/template-vision-dcgan/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
torch>=1.8.0
22
torchvision>=0.9.0
3-
pytorch-ignite>=0.4.4
3+
pytorch-ignite>=0.4.5
44
pyyaml
55

66
#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#

src/templates/template-vision-segmentation/main.py

+2-9
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
import ignite.distributed as idist
66
import yaml
77
from data import denormalize, setup_data
8-
from ignite.contrib.handlers import LRScheduler
98
from ignite.engine import Events
9+
from ignite.handlers import LRScheduler
1010
from ignite.metrics import ConfusionMatrix, IoU, mIoU
1111
from ignite.utils import manual_seed
1212
from 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) { :::#

src/templates/template-vision-segmentation/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
torch>=1.8.0
22
torchvision>=0.9.0
3-
pytorch-ignite>=0.4.4
3+
pytorch-ignite>=0.4.5
44
pyyaml
55
albumentations
66
image_dataset_viz

0 commit comments

Comments
 (0)