-
-
Notifications
You must be signed in to change notification settings - Fork 654
Closed
Description
CallableEventWithFilter
does not work with custom events:
from ignite.engine import Engine, EventEnum
def train_step(engine, batch):
engine.fire_event(TestEvents.TEST_EVENT)
engine = Engine(train_step)
class TestEvents(EventEnum):
TEST_EVENT = 'test_event'
engine.register_events(*TestEvents)
@engine.on(TestEvents.TEST_EVENT(every=100))
def foo(engine):
print("hello world")
def forever():
while True:
yield 0.
engine.run(forever(), max_epochs=10, epoch_length=1)
The error is:
RuntimeError: Unknown event name '<event=TEST_EVENT, filter=<function CallableEventWithFilter.every_event_filter.<locals>.wrapper at 0x7fb7d1244710>>'