File tree 10 files changed +12
-47
lines changed
template-text-classification
template-vision-classification
template-vision-segmentation
10 files changed +12
-47
lines changed Original file line number Diff line number Diff line change 4
4
5
5
#::: if (it.logger) { :::#
6
6
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 ()
15
8
#::: } :::#
16
9
17
10
#::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 1
1
torch >= 1.8.0
2
2
torchvision >= 0.9.0
3
- pytorch-ignite >= 0.4.4
3
+ pytorch-ignite >= 0.4.5
4
4
pyyaml
5
5
6
6
#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#
Original file line number Diff line number Diff line change 5
5
import ignite .distributed as idist
6
6
import yaml
7
7
from data import setup_data
8
- from ignite .contrib .handlers import LRScheduler , PiecewiseLinear
9
8
from ignite .engine import Events
9
+ from ignite .handlers import LRScheduler , PiecewiseLinear
10
10
from ignite .metrics import Accuracy , Loss
11
11
from ignite .utils import manual_seed
12
12
from models import TransformerModel
@@ -157,14 +157,7 @@ def _():
157
157
#::: if (it.logger) { :::#
158
158
# close logger
159
159
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 ()
168
161
#::: } :::#
169
162
#
170
163
#::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 1
1
torch >= 1.8.0
2
2
torchvision >= 0.9.0
3
- pytorch-ignite >= 0.4.4
3
+ pytorch-ignite >= 0.4.5
4
4
pyyaml
5
5
transformers
6
6
datasets
Original file line number Diff line number Diff line change @@ -113,14 +113,7 @@ def _():
113
113
#::: if (it.logger) { :::#
114
114
# close logger
115
115
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 ()
124
117
#::: } :::#
125
118
#
126
119
#::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 1
1
torch >= 1.8.0
2
2
torchvision >= 0.9.0
3
- pytorch-ignite >= 0.4.4
3
+ pytorch-ignite >= 0.4.5
4
4
pyyaml
5
5
6
6
#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#
Original file line number Diff line number Diff line change @@ -167,14 +167,7 @@ def _():
167
167
#::: if (it.logger) { :::#
168
168
# close logger
169
169
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 ()
178
171
#::: } :::#
179
172
#
180
173
#::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 1
1
torch >= 1.8.0
2
2
torchvision >= 0.9.0
3
- pytorch-ignite >= 0.4.4
3
+ pytorch-ignite >= 0.4.5
4
4
pyyaml
5
5
6
6
#::: if (['neptune', 'polyaxon'].includes(it.logger)) { :::#
Original file line number Diff line number Diff line change 5
5
import ignite .distributed as idist
6
6
import yaml
7
7
from data import denormalize , setup_data
8
- from ignite .contrib .handlers import LRScheduler
9
8
from ignite .engine import Events
9
+ from ignite .handlers import LRScheduler
10
10
from ignite .metrics import ConfusionMatrix , IoU , mIoU
11
11
from ignite .utils import manual_seed
12
12
from models import setup_model
@@ -176,14 +176,7 @@ def _():
176
176
#::: if (it.logger) { :::#
177
177
# close logger
178
178
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 ()
187
180
#::: } :::#
188
181
#
189
182
#::: if (it.save_training || it.save_evaluation) { :::#
Original file line number Diff line number Diff line change 1
1
torch >= 1.8.0
2
2
torchvision >= 0.9.0
3
- pytorch-ignite >= 0.4.4
3
+ pytorch-ignite >= 0.4.5
4
4
pyyaml
5
5
albumentations
6
6
image_dataset_viz
You can’t perform that action at this time.
0 commit comments