Path	Lines of Code
exps/eval_rm.py	34
exps/sample.py	43
setup.py	8
summarize_from_feedback/__init__.py	2
summarize_from_feedback/datasets/__init__.py	33
summarize_from_feedback/datasets/cnndm.py	121
summarize_from_feedback/datasets/encodings.py	144
summarize_from_feedback/datasets/jsonl_encoding.py	39
summarize_from_feedback/datasets/tldr.py	16
summarize_from_feedback/eval_rm.py	78
summarize_from_feedback/model_layout.py	46
summarize_from_feedback/models/__init__.py	1
summarize_from_feedback/models/attention.py	100
summarize_from_feedback/models/loss_functions.py	13
summarize_from_feedback/models/ops.py	41
summarize_from_feedback/models/sample_fns.py	39
summarize_from_feedback/models/transformer.py	429
summarize_from_feedback/policy.py	51
summarize_from_feedback/query_response_model.py	530
summarize_from_feedback/reward_model.py	42
summarize_from_feedback/sample.py	259
summarize_from_feedback/task_data.py	92
summarize_from_feedback/tasks.py	134
summarize_from_feedback/utils/__init__.py	15
summarize_from_feedback/utils/assertions.py	37
summarize_from_feedback/utils/blobs.py	57
summarize_from_feedback/utils/combos.py	58
summarize_from_feedback/utils/dist_utils.py	141
summarize_from_feedback/utils/even_more_itertools.py	13
summarize_from_feedback/utils/experiments.py	66
summarize_from_feedback/utils/hyperparams.py	198
summarize_from_feedback/utils/jobs.py	37
summarize_from_feedback/utils/logging_utils.py	17
summarize_from_feedback/utils/nested.py	23
summarize_from_feedback/utils/torch_utils.py	36