Hi @mpouzinis, the variables defined in __init__() should be persisted within the same round (see here, e.g., self.device or self.model persists and can be called in fit() and evaluate()). Could you provide more context or code regarding the issues you encountered?
2 Likes