plot images on cpu

[hvgazula]

tissue_labeling/TissueLabeling/training/trainer.py

Lines 35 to 39 in 59f5ea9

	if self.fabric.global_rank == 0:
	self.image_logger = Log_Images(
	self.fabric,
	config=config,
	)

(optional) it doesn't make sense to plot images (or have that functionality) in the trainer. plot images using cpu and then call trainer.