[Torch] Fix classification sample metric dumps not in main process

[Torch] Fix classification sample metric dumps not in main process #414