/LSUV-keras

Simple implementation of the LSUV initialization in keras

Primary LanguagePythonBSD 2-Clause "Simplified" LicenseBSD-2-Clause

Layer-sequential unit-variance (LSUV) initialization for Keras

This is sample code for LSUV and initializations, implemented in python script within Keras framework.

Usage:

from lsuv_init import LSUVinit
...
batch_size = 32
model = LSUVinit(model, train_imgs[:batch_size,:,:,:])

LSUV initialization is described in:

Mishkin, D. and Matas, J.,(2015). All you need is a good init. ICLR 2016 arXiv:1511.06422.

Original Caffe implementation https://github.com/ducha-aiki/LSUVinit

Torch re-implementation https://github.com/yobibyte/torch-lsuv

PyTorch implementation https://github.com/ducha-aiki/LSUV-pytorch

New! Thinc re-implementation LSUV-thinc