Skip to content

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

simplify

parent 17a4228a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
examples/DoReFa-Net/alexnet-dorefa.py
View file @ 300b2c3a
...@@ -155,8 +155,7 @@ class Model(ModelDesc): ...@@ -155,8 +155,7 @@ class Model(ModelDesc):
def get_data(dataset_name): def get_data(dataset_name):
isTrain = dataset_name == 'train' isTrain = dataset_name == 'train'
ds = dataset.ILSVRC12(args.data, dataset_name, ds = dataset.ILSVRC12(args.data, dataset_name, shuffle=isTrain)
shuffle=True if isTrain else False)
meta = dataset.ILSVRCMeta() meta = dataset.ILSVRCMeta()
pp_mean = meta.get_per_pixel_mean() pp_mean = meta.get_per_pixel_mean()
......
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