def initialize_states(self, states): states.units.source = ListEntry() states.units.target = ListEntry() states.segments.source = ListEntry() states.segments.target = ListEntry() states.unit_queue.source = QueueEntry() states.unit_queue.target = QueueEntry() states.encoder_cache = [{} for _ in self.models] states.decoder_cache = [{} for _ in self.models] states.segment = False states.encoding_time, states.decoding_time = 0, 0
def initialize_states(self, states): self.feature_extractor.clear_cache() states.units.source = TensorListEntry() states.units.target = ListEntry() states.incremental_states = dict()