Searched refs:train_state_manager (Results 1 – 1 of 1) sorted by relevance
107 train_state_manager=state_management.PassthroughStateManager()): argument155 state_manager=train_state_manager,176 train_state_manager=state_management.PassthroughStateManager()): argument218 train_state_manager=train_state_manager)259 train_state_manager=state_management.PassthroughStateManager()): argument281 train_state_manager=train_state_manager,