Skip to content
This repository has been archived by the owner on Jul 2, 2021. It is now read-only.

Add fine-tuning tutorial #601

Merged
merged 2 commits into from
May 10, 2018
Merged

Add fine-tuning tutorial #601

merged 2 commits into from
May 10, 2018

Conversation

yuyu2172
Copy link
Member

Related
#416

# the number of classes in VOC is different from 50
dst = SSD300(n_fg_class=50)
# initialized weights
dst(np.zeros((1, 3, 300, 300)).astype(np.float32))
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

How about using model.insize instead of 300?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You can write np.zeros((1, 3, model.insize, insize), dtype=np.float32).

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

sure

@yuyu2172 yuyu2172 added this to the 0.10 milestone May 10, 2018
Copy link
Member

@Hakuyume Hakuyume left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@Hakuyume Hakuyume merged commit d2c83e8 into chainer:master May 10, 2018
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants