Allow profiling to be toggled using a config flag instead of adding/removing a decorator

[jarlsondre]

Summary

If you wish to turn itwinai's profiling functionality on or off, you currently have to add or remove a decorator. This is quite cumbersome, as one shouldn't have to manually change the code just to toggle a configuration setting. Additionally, it is problematic as you also have to remove the import of the decorator in most cases—at least for the linter not to fail.

Potential Solution: This can be solved by adding a field in the TorchTrainer that the decorator can use to decide whether or not to activate. This field can easily be passed to the TorchTrainer from a configuration, thus solving the problem.