Open exx8 opened 1 year ago
add parallelism to the classification training.
Hi, have you tested if the performance keeps the same or similar, since dataparallel may affect the batchnorm behavior? See https://discuss.pytorch.org/t/how-does-dataparallel-handels-batch-norm/14040
add parallelism to the classification training.