Skip to content

S
seminar-breakout
Commit c2c895ab authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

clean-ups in mnist-keras-v2

parent 59d099cd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 6 deletions
examples/mnist-keras-v2.py
View file @ c2c895ab
...@@ -5,12 +5,7 @@ ...@@ -5,12 +5,7 @@
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
import tensorflow.contrib.slim as slim
import os
import sys
import argparse
import keras
from keras.models import Sequential from keras.models import Sequential
import keras.layers as KL import keras.layers as KL
from keras import regularizers from keras import regularizers
...@@ -18,7 +13,7 @@ from keras import regularizers ...@@ -18,7 +13,7 @@ from keras import regularizers
from tensorpack.train import SimpleTrainer from tensorpack.train import SimpleTrainer
from tensorpack.input_source import QueueInput from tensorpack.input_source import QueueInput
from tensorpack.callbacks import * from tensorpack.callbacks import ModelSaver, InferenceRunner, ScalarStats
from tensorpack.dataflow import dataset, BatchData, MapData from tensorpack.dataflow import dataset, BatchData, MapData
from tensorpack.utils import logger from tensorpack.utils import logger
from tensorpack.contrib.keras import setup_keras_trainer from tensorpack.contrib.keras import setup_keras_trainer
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment