Sadcardation

Same problem! Training on TITAN RTX, with very low CPU and GPU Utilization.

> > Besides, compute_loss required 3 parameters, so we have to remove "imgs" params in the input, does it affect the training? > > This is correct. > > I...