marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-11 06:15:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

renamed sgd.h to optimizers.h

Browse Source
This commit is contained in:
Marcin Junczys-Dowmunt 2016-09-16 23:55:47 +02:00
parent 6e90198426
commit 09ce7e6349
3 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
src/sgd.h → src/optimizers.h
View File

10
src/train_mnist.cu
View File

@ -1,17 +1,15 @@
#include "marian.h"
#include "mnist.h"
#include "sgd.h"
using namespace std;
#include "optimizers.h"
int main(int argc, char** argv) {
const size_t IMAGE_SIZE = 784;
const size_t LABEL_SIZE = 10;
int numofdata;
vector<float> trainImages = datasets::mnist::ReadImages("../examples/mnist/t10k-images-idx3-ubyte", numofdata, IMAGE_SIZE);
vector<float> trainLabels = datasets::mnist::ReadLabels("../examples/mnist/t10k-labels-idx1-ubyte", numofdata, LABEL_SIZE);
std::vector<float> trainImages = datasets::mnist::ReadImages("../examples/mnist/t10k-images-idx3-ubyte", numofdata, IMAGE_SIZE);
std::vector<float> trainLabels = datasets::mnist::ReadLabels("../examples/mnist/t10k-labels-idx1-ubyte", numofdata, LABEL_SIZE);
using namespace marian;
using namespace keywords;
@ -27,7 +25,7 @@ int main(int argc, char** argv) {
auto scores = dot(x, w) + b;
auto lr = softmax(scores);
auto cost = named(-mean(sum(y * log(lr), axis=1), axis=0), "cost");
cerr << "lr=" << lr.Debug() << endl;
std::cerr << "lr=" << lr.Debug() << std::endl;
Adagrad opt;
opt(g, 300);

2
src/validate_mnist.cu
View File

@ -2,7 +2,7 @@
#include "marian.h"
#include "mnist.h"
#include "npz_converter.h"
#include "sgd.h"
#include "optimizers.h"
using namespace marian;
using namespace keywords;