Skip to content

Instantly share code, notes, and snippets.

View piiswrong's full-sized avatar

Eric Junyuan Xie piiswrong

1.2k followers · 22 following
View GitHub Profile
@piiswrong
piiswrong / test_operator.py
Last active October 24, 2015 22:52
Engine Crashes if asnumpy() is not called on result
import os
import mxnet as mx
from common import models
import pickle as pkl
def test_softmax_with_shape(shape, xpu):
X = mx.symbol.Variable('X')
L = mx.symbol.Variable('L')
Y = mx.symbol.Softmax(data=X, label=L)
x = mx.random.uniform(-1, 1, shape, ctx = xpu)
@piiswrong
piiswrong / prod_sum-inl.h
Created December 19, 2015 21:21
/*!
* Copyright (c) 2015 by Contributors
* \file leaky_relu-inl.h
* \brief leaky relu family operator
* \author Bing Xu
*/
#ifndef MXNET_OPERATOR_PROD_SUM_INL_H_
#define MXNET_OPERATOR_PROD_SUM_INL_H_
#include <dmlc/logging.h>
@piiswrong
piiswrong / gist:d859ba805607775a8306d310e0e8345c
Created September 26, 2016 22:57
def fit(args, network, data_loader, batch_end_callback=None):
# kvstore
kv = mx.kvstore.create(args.kv_store)
model_prefix = args.model_prefix
if model_prefix is not None:
model_prefix += "-%d" % (kv.rank)
save_model_prefix = args.save_model_prefix
if save_model_prefix is None:
@piiswrong
piiswrong / install.sh
Last active December 12, 2017 07:58
install mxnet on ubuntu
#!/usr/bin/env bash
set -e
# install cuda-7.5
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7.5-18_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1404_7.5-18_amd64.deb
sudo apt-get update
sudo apt-get install -y linux-image-extra-`uname -r` linux-headers-`uname -r` linux-image-`uname -r`
sudo apt-get install -y cuda-7-5
echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:\$LD_LIBRARY_PATH" | tee -a ~/.profile | tee -a ~/.bashrc
@piiswrong
piiswrong / mxnet_predict-all.cc
Created January 25, 2017 21:30
This file has been truncated, but you can view the full file.
#if defined(__MACH__)
#include <mach/clock.h>
#include <mach/mach.h>
#endif
#if !defined(__WIN32__)
#include <sys/stat.h>
#include <sys/types.h>
@piiswrong
piiswrong / foobar.ipynb
Last active July 26, 2017 23:00
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.