I was replicating a paper that has a complex model built in pytorch and wanted it in tensorflow for ease of use (I have been working for a while with tensorflow seems user friendly more for me please don’t punch me), anyways now i run into some problems with the subclassed model it trains fine and everything but some problems with saving and loading and also types of save (.h5, etc…) and i also would like to quantize it later on either by QAT or quantization after training and it seems doable/easier to do in functional without all the errors i have been getting.