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 {
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",
18401841 },
18411842 "nbformat": 4,
18421843 "nbformat_minor": 2
1843- }
1844+ }
0 commit comments