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:
![image](https://private-user-images.githubusercontent.com/131862196/293195662-cab49784-31a8-42ff-958d-b6a931ef4276.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjExNDMxNjAsIm5iZiI6MTcyMTE0Mjg2MCwicGF0aCI6Ii8xMzE4NjIxOTYvMjkzMTk1NjYyLWNhYjQ5Nzg0LTMxYTgtNDJmZi05NThkLWI2YTkzMWVmNDI3Ni5wbmc_WC1BbXotQWxnb3JpdGhtPUFXUzQtSE1BQy1TSEEyNTYmWC1BbXotQ3JlZGVudGlhbD1BS0lBVkNPRFlMU0E1M1BRSzRaQSUyRjIwMjQwNzE2JTJGdXMtZWFzdC0xJTJGczMlMkZhd3M0X3JlcXVlc3QmWC1BbXotRGF0ZT0yMDI0MDcxNlQxNTE0MjBaJlgtQW16LUV4cGlyZXM9MzAwJlgtQW16LVNpZ25hdHVyZT1mZmExMzU2MWQwN2FkZTFjMjYzZTkzMTlmMjYwNGIxMjViM2I1MTg0NmFjOTdmM2Y2OWYyMGUzYjRhOTUyYmQzJlgtQW16LVNpZ25lZEhlYWRlcnM9aG9zdCZhY3Rvcl9pZD0wJmtleV9pZD0wJnJlcG9faWQ9MCJ9.1gTy5fRWMZZcedZaxsHN35WUfx3blaaA1j_1mDNgxN4)
Missing Required Argument in Callbacks Constructor
Another TypeError I encountered is related to the instantiation of the Callbacks class:
![image](https://private-user-images.githubusercontent.com/131862196/293195610-f7a34256-a5ef-4f9c-83ae-b7dd4f219dd9.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjExNDMxNjAsIm5iZiI6MTcyMTE0Mjg2MCwicGF0aCI6Ii8xMzE4NjIxOTYvMjkzMTk1NjEwLWY3YTM0MjU2LWE1ZWYtNGY5Yy04M2FlLWI3ZGQ0ZjIxOWRkOS5wbmc_WC1BbXotQWxnb3JpdGhtPUFXUzQtSE1BQy1TSEEyNTYmWC1BbXotQ3JlZGVudGlhbD1BS0lBVkNPRFlMU0E1M1BRSzRaQSUyRjIwMjQwNzE2JTJGdXMtZWFzdC0xJTJGczMlMkZhd3M0X3JlcXVlc3QmWC1BbXotRGF0ZT0yMDI0MDcxNlQxNTE0MjBaJlgtQW16LUV4cGlyZXM9MzAwJlgtQW16LVNpZ25hdHVyZT02MjA0NTMwYjE1OWQ4NmViOWI3MmJmMDBmNjkxMTNkOWU2YTJjZjg0NjljZTFkNjYyODA3MGVmNGI3MzJiZTgxJlgtQW16LVNpZ25lZEhlYWRlcnM9aG9zdCZhY3Rvcl9pZD0wJmtleV9pZD0wJnJlcG9faWQ9MCJ9.Yd-z1L1ZTEFn8JYaCJH1nUED0xEgkdCX2Euc7V7CdqU)
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.