I am facing problem with implementing WandB for tracking my RL training algorithm. I am using Keras for writing my training code. However, I do not use the model.fit function. I am instead using @tf.function to perform updates on the Q-network and the Policy-network (I am using DDPG learning algorithm). How can I initialize a run and use the tracking callbacks if I am not using the model.fit so I cannot pass the callback as an argument. Every guide/tutorial on the internet for using Keras with WandB assumes that I am using model.fit and this is not the case. Can you help please! Thanks a lot for your help.
I have not tried anything till now because I have no clue where to start. I could not find anyone on the internet with a similar code or problem.