Loading request...
The current implementation uses default parameters for AdamW. The user requests the ability to configure optimizer parameters and suggests having a standalone optimizer class with a registry for more flexibility.
Currently use default parameters of AdamW: https://github.com/salesforce/LAVIS/blob/main/lavis/runners/runner_base.py#L103. This has to be made more flexible. Also, it would be good to have a standalone optimizer class, and related registry etc.