utils.py (1 line):
	- line 104: # TODO: avoid this so that distributed cpu training also works