Skip to content

Commit

Permalink
Fix imagenet training script with data loader (#184)
Browse files Browse the repository at this point in the history
* in initial structure

* fix

* improve verification

* slight modification

* update tutorial

* prepare recio

* fix training script

* fix tutorial

* update scheduler to include warmup, cosine and step

* update tutorials

* fix

* fix

* fix lint

* Trigger CI

* improve doc

* change save_params to save_parameters

* temporary download test files

* update download command

* fix

* add imagenet mixup

* fix

* fix

* update training results

* remove test commands in jenkins

* fix tutorial

* fix

* Trigger CI

* Trigger CI

* download files for tutorial

* update

* fix

* remove download scripts, waiting for rec file to be added

* retry download

* Trigger CI

* remove download commands

* including download, update required pip version

* download for doc building

* fix path

* finalize tutorial shape, hold for pr in mxnet repo

* update resnet152 results

* need the nightly built mxnet

* fix yml

* fix doc

* fix doc

* fix doc?

* rename class, fix doc

* fix

* improve

* fix doc

* fix training script

* fix
  • Loading branch information
hetong007 authored Jun 25, 2018
1 parent 25b1717 commit e915c0b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion scripts/classification/imagenet/train_imagenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -197,7 +197,7 @@ def batch_fn(batch, ctx):
normalize
])
transform_test = transforms.Compose([
transforms.Resize(256),
transforms.Resize(256, keep_ratio=True),
transforms.CenterCrop(224),
transforms.ToTensor(),
normalize
Expand Down

0 comments on commit e915c0b

Please sign in to comment.