Giving methods to updater is not possible

flowersteam / lamorel

Lamorel is a Python library designed for RL practitioners eager to use Large Language Models (LLMs).

MIT License

190 stars 18 forks source link

Giving methods to updater is not possible #1

Closed ClementRomac closed 1 year ago

ClementRomac commented 1 year ago

Giving functions (e.g. lambdas) to the updater would be nice. As it is not possible to do it on perform_update (everything go through torch dist), the constructor should be accessible.