Skip to content
Snippets Groups Projects
Commit 92f3cb71 authored by Olivier Canévet's avatar Olivier Canévet
Browse files

Add test_lstm

parent 067c701f
Branches
No related tags found
No related merge requests found
#!/usr/bin/env python
import numpy
from bob.learn.tensorflow.datashuffler import Memory, ScaleFactor
from bob.learn.tensorflow.network import MLP, Embedding
from bob.learn.tensorflow.loss import BaseLoss
from bob.learn.tensorflow.trainers import Trainer, constant
from bob.learn.tensorflow.utils import load_mnist
import tensorflow as tf
train_data, train_labels, validation_data, validation_labels = load_mnist(data_dir="mnist")
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment