Update engine.py to avoid torch warning (#5408)
The state_dict function of module.py from torch write a warning if
arguments are positional arguments and not keyword arguments
---------
Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Co-authored-by: ebonnafoux <ebonnafoux156@headmind.com>
