complete the code here

[todo]

Contrast-You/contrastyou/epocher/contrast_epocher.py

Lines 30 to 35 in 2086023

	# todo: complete the code here
	pass
	def _run(self, *args, **kwargs) -> EpochResultDict:
	self._model.set_mode(ModelMode.TRAIN)
	assert self._model.training, self._model.training

---

This issue was generated by todo based on a todo comment in 2086023. It's been assigned to @jizongFox because they committed the code.