Skip to content

Commit f521836

Browse files
committed
Remove deprecated CustomPeriodicEvent from nb example, fix tb OutputHadler
1 parent 5dacbdf commit f521836

File tree

1 file changed

+20
-19
lines changed

1 file changed

+20
-19
lines changed

examples/notebooks/EfficientNet_Cifar100_finetuning.ipynb

Lines changed: 20 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1429,34 +1429,35 @@
14291429
"metadata": {},
14301430
"outputs": [],
14311431
"source": [
1432-
"from ignite.contrib.handlers import CustomPeriodicEvent\n",
1433-
"\n",
1434-
"cpe = CustomPeriodicEvent(n_epochs=3)\n",
1435-
"cpe.attach(trainer)\n",
1432+
"from ignite.handlers import global_step_from_engine\n",
14361433
"\n",
14371434
"\n",
14381435
"def run_evaluation(engine):\n",
14391436
" train_evaluator.run(eval_train_loader)\n",
14401437
" evaluator.run(test_loader)\n",
14411438
"\n",
14421439
"\n",
1443-
"trainer.add_event_handler(cpe.Events.EPOCHS_3_STARTED, run_evaluation)\n",
1440+
"trainer.add_event_handler(Events.EPOCH_STARTED(every=3), run_evaluation)\n",
14441441
"trainer.add_event_handler(Events.COMPLETED, run_evaluation)\n",
14451442
"\n",
14461443
"\n",
14471444
"# Log train eval metrics:\n",
1448-
"tb_logger.attach(train_evaluator,\n",
1449-
" log_handler=OutputHandler(tag=\"training\",\n",
1450-
" metric_names=list(metrics.keys()),\n",
1451-
" another_engine=trainer),\n",
1452-
" event_name=Events.EPOCH_COMPLETED)\n",
1445+
"tb_logger.attach_output_handler(\n",
1446+
" train_evaluator,\n",
1447+
" event_name=Events.EPOCH_COMPLETED\n",
1448+
" tag=\"training\",\n",
1449+
" metric_names=list(metrics.keys()),\n",
1450+
" global_step_transform=global_step_from_engine(trainer)\n",
1451+
")\n",
14531452
"\n",
14541453
"# Log val metrics:\n",
1455-
"tb_logger.attach(evaluator,\n",
1456-
" log_handler=OutputHandler(tag=\"test\",\n",
1457-
" metric_names=list(metrics.keys()),\n",
1458-
" another_engine=trainer),\n",
1459-
" event_name=Events.EPOCH_COMPLETED)"
1454+
"tb_logger.attach_output_handler(\n",
1455+
" evaluator,\n",
1456+
" event_name=Events.EPOCH_COMPLETED,\n",
1457+
" tag=\"test\",\n",
1458+
" metric_names=list(metrics.keys()),\n",
1459+
" global_step_transform=global_step_from_engine(trainer)\n",
1460+
")"
14601461
]
14611462
},
14621463
{
@@ -1513,7 +1514,7 @@
15131514
"es_patience = 10\n",
15141515
"es_handler = EarlyStopping(patience=es_patience, score_function=default_score_fn, trainer=trainer)\n",
15151516
"evaluator.add_event_handler(Events.COMPLETED, es_handler)\n",
1516-
"setup_logger(es_handler._logger)\n",
1517+
"setup_logger(es_handler.logger)\n",
15171518
"\n",
15181519
"\n",
15191520
"# Clear cuda cache between training/testing\n",
@@ -1835,9 +1836,9 @@
18351836
"name": "python",
18361837
"nbconvert_exporter": "python",
18371838
"pygments_lexer": "ipython3",
1838-
"version": "3.7.4"
1839+
"version": "3.7.6"
18391840
}
18401841
},
18411842
"nbformat": 4,
1842-
"nbformat_minor": 2
1843-
}
1843+
"nbformat_minor": 4
1844+
}

0 commit comments

Comments
 (0)