I have been trying to make a custom training function for my models and have stumbled upon an error that I can’t seem to fix nor understand. When trying to update my weights using the method optimize.apply_gradients, I get the following error :
“Type Error : tuple indices must be integers or slices, not MirroredVariable”
I assume that this is due to my models being originally created using a mirrored strategy to use multiple gpus and dispatch the work load. Is there a way to work around this issue ?