TypeError in pt_train.fit and pt_callbacks.Callbacks Calls

[SHChang-Anderson]

Hello,

I've encountered a couple of TypeErrors while using the fit function from source/pytorch_utils/training_utils.py and Callbacks class from pt_callbacks. Here are the details of the issues:

Unexpected Keyword Argument in get_callbacks Function

While running pt_train.fit, I encountered the following TypeError:

Missing Required Argument in Callbacks Constructor

Another TypeError I encountered is related to the instantiation of the Callbacks class:

It seems that there might be a mismatch between the expected parameters in the get_callbacks function and the arguments passed to it from pt_train.fit. Also, the Callbacks class appears to require an optimizer argument, which is not being provided in the get_callbacks function.

Could you please help in resolving these issues? Is there a specific way these functions are intended to be used, or is this potentially a bug in the implementation?

Thank you for your assistance.