Hello ! thank you for your code.but i meet problem:
AttributeError: 'NoneType' object has no attribute 'run_slave'

and i set batch_size=8 ,i only have one gpu.