Solved by Ohm in post #8 
              
              
                A simple example to split dataset among n_workers: 
import torch 
import torch.nn as nn 
import torchvision 
import torchvision.transforms as transforms 
import numpy as np 
import time 
from worker import worker 
from worker_new import worker_new 
#the number of workers here: 
n_workers = 4 
testse…