Fix init_fn in train_utils. (#7790)
Fix init_fn in train_utils. (#7790)


alexgorban/models,alexgorban/models,alexgorban/models,alexgorban/models,tombstone/models,tombstone/models,tombstone/models,tombstone/models,alexgorban/models,tombstone/models,tombstone/models