报错语句处:
assert (n_batch % (n_gpus - 1) == 0), 'n_batch: {}'.format(n_batch)
*无论我n_batch输入是什么,都报错n_batch=xxx出错。*
首先我们理解一下assert语句是module文件下的,指我们希望得到的是assert内部的情况,否则就报错。
那这里就是需要n_batch % (n_gpus - 1) == 0
,如果n_gpus - 1=8,那么n_batch只要是7的倍数就可以啦!~