* Enable to toggle whether Gem Pooling is trainable or not. * Add test case whether Gem Pooling is trainable or not. * Enable to toggle whether Gem Pooling is trainable or not by requires_grad --------- Co-authored-by: Yusuke Fujimoto <yusuke.fujimoto@rist.co.jp> |
||
---|---|---|
.. | ||
__init__.py | ||
gap.py | ||
gem.py | ||
hr_fuse.py | ||
reduction.py |