Commit 73ac38c7 authored by Yuxin Wu's avatar Yuxin Wu

misc.

parent 90a74f02
# tensorpack # tensorpack
Neural Network Toolbox on TensorFlow Neural Network Toolbox on TensorFlow
In development. API might change a bit. See [examples](https://github.com/ppwwyyxx/tensorpack/tree/master/examples) to learn. In development. API might change a bit.
See [examples](https://github.com/ppwwyyxx/tensorpack/tree/master/examples) to learn.
## Features: ## Features:
+ Scoped abstraction of common models.
+ Use `Dataflow` to define data preprocessing in pure Python. Focused on modularity:
+ Callbacks systems to control training.
+ Training and testing are described together. Just need to follow the conventions to setup stuffs. + Models has Scoped abstraction of common models.
+ Write summary easier for tensorboard. + Dataflow defines data preprocessing in pure Python.
+ Callbacks systems to control training behavior.
...@@ -21,6 +21,12 @@ from tensorpack.callbacks import * ...@@ -21,6 +21,12 @@ from tensorpack.callbacks import *
from tensorpack.dataflow import * from tensorpack.dataflow import *
from tensorpack.dataflow.dataset import ILSVRCMeta from tensorpack.dataflow.dataset import ILSVRCMeta
"""
Usage:
python2 -m tensorpack.utils.loadcaffe PATH/TO/CAFFE/{deploy.prototxt,bvlc_alexnet.caffemodel} alexnet.npy
./load-alexnet.py --load alexnet.npy --input cat.png
"""
BATCH_SIZE = 10 BATCH_SIZE = 10
MIN_AFTER_DEQUEUE = 500 MIN_AFTER_DEQUEUE = 500
CAPACITY = MIN_AFTER_DEQUEUE + 3 * BATCH_SIZE CAPACITY = MIN_AFTER_DEQUEUE + 3 * BATCH_SIZE
......
...@@ -23,8 +23,8 @@ from tensorpack.dataflow.dataset import ILSVRCMeta ...@@ -23,8 +23,8 @@ from tensorpack.dataflow.dataset import ILSVRCMeta
""" """
Usage: Usage:
python2 -m tensorpack.utils.loadcaffe PATH/TO/models/VGG/{VGG_ILSVRC_16_layers_deploy.prototxt,VGG_ILSVRC_16_layers.caffemodel} vgg16.npy python2 -m tensorpack.utils.loadcaffe PATH/TO/VGG/{VGG_ILSVRC_16_layers_deploy.prototxt,VGG_ILSVRC_16_layers.caffemodel} vgg16.npy
./load_vgg16.py --load vgg16.npy --input cat.png ./load-vgg16.py --load vgg16.npy --input cat.png
""" """
class Model(ModelDesc): class Model(ModelDesc):
......
...@@ -168,10 +168,8 @@ class AtariPlayer(RLEnvironment): ...@@ -168,10 +168,8 @@ class AtariPlayer(RLEnvironment):
def get_stat(self): def get_stat(self):
try: try:
print self.stats
return {'avg_score': np.mean(self.stats['score']), return {'avg_score': np.mean(self.stats['score']),
'max_score': float(np.max(self.stats['score'])) 'max_score': float(np.max(self.stats['score'])) }
}
except ValueError: except ValueError:
return {} return {}
......
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