Gpu is almost not being used while training but data and model are on device

20

@Jaredeco Did you try playing with the num_workers argument :point_right: Guidelines for assigning num_workers to DataLoader

Other than that, I don’t really have much to add. I hope your issue is resolved quickly :+1:.