Skip to content

Commit 11f6d62

Browse files
authored
Remove deprecated CustomPeriodicEvent from nb example, fix tb OutputHadler (#1389)
1 parent 5dacbdf commit 11f6d62

File tree

1 file changed

+18
-17
lines changed

1 file changed

+18
-17
lines changed

examples/notebooks/EfficientNet_Cifar100_finetuning.ipynb

Lines changed: 18 additions & 17 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",
@@ -1840,4 +1841,4 @@
18401841
},
18411842
"nbformat": 4,
18421843
"nbformat_minor": 2
1843-
}
1844+
}

0 commit comments

Comments
 (0)