Fix LR scheduler issue on CI #232
Description
Describe the bug
Currently CI is failing on segmenation example:
[ignite]: Configuration:
{'accumulation_steps': 4,
'backend': None,
'data_path': '/home/runner/data',
'debug': False,
'eval_batch_size': 2,
'eval_epoch_length': 4,
'filename_prefix': 'training',
'limit_sec': 60,
'log_every_iters': 2,
'lr': 0.007,
'max_epochs': 2,
'n_saved': 2,
'num_classes': 21,
'num_workers': 2,
'output_dir': PosixPath('logs/20230320-090125-backend-None-lr-0.007'),
'patience': 2,
'save_every_iters': 2,
'seed': 666,
'train_batch_size': 2,
'train_epoch_length': 4,
'use_amp': False}
Traceback (most recent call last):
File "/home/runner/work/code-generator/code-generator/dist-tests/vision-segmentation-all/main.py", line 181, in <module>
main()
File "/home/runner/work/code-generator/code-generator/dist-tests/vision-segmentation-all/main.py", line 177, in main
p.run(run, config=config)
File "/opt/hostedtoolcache/Python/3.9.16/x64/lib/python3.9/site-packages/ignite/distributed/launcher.py", line 316, in run
func(local_rank, *args, **kwargs)
File "/home/runner/work/code-generator/code-generator/dist-tests/vision-segmentation-all/main.py", line 83, in run
trainer.add_event_handler(Events.ITERATION_STARTED, lr_scheduler)
File "/opt/hostedtoolcache/Python/3.9.16/x64/lib/python3.9/site-packages/ignite/engine/engine.py", line 319, in add_event_handler
_check_signature(handler, "handler", self, *(event_args + args), **kwargs)
File "/opt/hostedtoolcache/Python/3.9.16/x64/lib/python3.9/site-packages/ignite/engine/utils.py", line 10, in _check_signature
signature = inspect.signature(fn)
File "/opt/hostedtoolcache/Python/3.9.16/x64/lib/python3.9/inspect.py", line 3113, in signature
return Signature.from_callable(obj, follow_wrapped=follow_wrapped)
File "/opt/hostedtoolcache/Python/3.9.16/x64/lib/python3.9/inspect.py", line 2862, in from_callable
return _signature_from_callable(obj, sigcls=cls,
File "/opt/hostedtoolcache/Python/3.9.16/x64/lib/python3.9/inspect.py", line 2261, in _signature_from_callable
raise TypeError('{!r} is not a callable object'.format(obj))
TypeError: <torch.optim.lr_scheduler.LambdaLR object at 0x7fdc0baab0a0> is not a callable object
Error: Process completed with exit code 1.