Add 'with EventStorage()' context

[ostrowskimarcin]

Currently I'm using pytorch to test some models benchmarks with compile mode - https://github.com/pytorch/pytorch/tree/main/benchmarks/dynamo
When I run detectron2-* models following error appears:

storage = get_event_storage()
  File "/usr/local/lib/python3.10/dist-packages/detectron2/utils/events.py", line 34, in get_event_storage
    assert len(
AssertionError: get_event_storage() has to be called inside a 'with EventStorage(...)' context!

It seems that the code doesn't get the storage in a proper way. I added a context to each required place and it works for me.
I also noticed that this usage is present in other different places in repo but I left it as it is.
Could you take a look?