search

glample / tagger

Named Entity Recognition Tool
Apache License 2.0
1.16k stars 426 forks source link

Bug fix: cannot train with RMSprop #19

Closed YoannDupont closed 6 years ago

YoannDupont commented 8 years ago

Issue

When trying to train with RMSprop learning method, Theano raised a TypeError.

Fix

Changed optimization.py line 157 :

accumulators = [theano.shared(np.zeros_like(p.get_value()).astype(np.float32)) for p in params]

To:

accumulators = [theano.shared(np.zeros_like(p.get_value()).astype(floatX)) for p in params]
glample commented 6 years ago

Thank you!