|
|
|
@ -17,9 +17,9 @@ import numpy as np
|
|
|
|
|
import contextlib
|
|
|
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
|
'Constant', 'Uniform', 'Normal', 'Xavier', 'MSRA', 'force_init_on_cpu',
|
|
|
|
|
'Constant', 'Uniform', 'Normal', 'Xavier', 'force_init_on_cpu',
|
|
|
|
|
'init_on_cpu', 'ConstantInitializer', 'UniformInitializer',
|
|
|
|
|
'NormalInitializer', 'XavierInitializer', 'MSRAInitializer'
|
|
|
|
|
'NormalInitializer', 'XavierInitializer'
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
_force_init_on_cpu_ = False
|
|
|
|
|