Skip to content

Instantly share code, notes, and snippets.

@feevos
feevos / gluon_accuracy_example_multi_gpu.ipynb
Created October 10, 2018 02:57
Example of multi-gpu accuracy evaluation
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@feevos
feevos / gluon_accuracy_example_multi_gpu.py
Last active October 10, 2018 14:55
example of mxnet/gluon accuracy evaluation on single/multi gpu context
from __future__ import print_function
import numpy as np
import mxnet as mx
from mxnet import nd, autograd, gluon
from time import time
import warnings
warnings.filterwarnings('ignore')
mx.random.seed(1)
batch_size = 256