INNER CODE UNIT · Python

inp

lucidrains/x-transformers · train_belief_state.py:126

        inp = random.choice(val_dataset)[:-1]
        prime = decode_tokens(inp)

        print(f'%s \n\n %s', (prime, '*' * 100))

        print('forwards:\n')

        sample = model.generate_with_suffix_cond(
            prompts = inp,
            seq_len = GENERATE_LENGTH,
            cache_kv = True
        )

        output_str = decode_tokens(sample)
        print(output_str)

        print('\nbackwards:\n')

View source record →