What if you want to output recurrent_states at all time steps, that is, each amino acid corresponds to a recurrent_states file? @alquraishi
What if you want to output recurrent_states at all time steps, that is, each amino acid corresponds to a recurrent_states file? @alquraishi