.\ |

vfdev vfdev-5

.\ |

contact: vfdev dot 5 at google mail

vfdev-5 / benchmark.py

Created September 6, 2020 20:36 — forked from n2cholas/benchmark.py

Benchmarking ignite master branch vs metrics_impl on metrics.

	'''
	To run the CPU benchmark: `CUDA_VISIBLE_DEVICES="" python benchmark.py --name cpu`
	To run the GPU benchmark: `CUDA_VISIBLE_DEVICES=0 python benchmark.py --name cuda`
	To run the distributed benchmark: `python -u -m torch.distributed.launch --nproc_per_node=2 --use_env benchmark.py --name dist`
	'''


	import argparse
	import time
	import math