However, in TF2, we can avoid doing this on the R side and do it completely in TF with. This would save us going back and forth between R and TF in the warmup phase, buying us some efficiency!
We can also explore tracking the progress bar in TF rather than from the R side every 50 iterations
For example, during warmup we step between R and TF every 4 iterations:
https://github.com/greta-dev/greta/blob/master/R/inference_class.R#L376
Using this code:
https://github.com/greta-dev/greta/blob/master/R/inference_class.R#L552-L622
However, in TF2, we can avoid doing this on the R side and do it completely in TF with. This would save us going back and forth between R and TF in the warmup phase, buying us some efficiency!
We can also explore tracking the progress bar in TF rather than from the R side every 50 iterations