Path	Lines of Code
common/__init__.py	1
common/batch.py	65
common/checkpointing/__init__.py	1
common/checkpointing/snapshot.py	173
common/device.py	23
common/filesystem/__init__.py	1
common/filesystem/util.py	15
common/log_weights.py	63
common/modules/embedding/config.py	34
common/modules/embedding/embedding.py	45
common/run_training.py	69
common/testing_utils.py	28
common/utils.py	23
common/wandb.py	15
core/__init__.py	1
core/config/__init__.py	3
core/config/base_config.py	32
core/config/config_load.py	13
core/config/training.py	32
core/custom_training_loop.py	235
core/debug_training_loop.py	25
core/loss_type.py	4
core/losses.py	74
core/metric_mixin.py	55
core/metrics.py	91
core/train_pipeline.py	421
machines/environment.py	78
machines/get_env.py	31
machines/is_venv.py	13
machines/list_ops.py	17
metrics/__init__.py	3
metrics/aggregation.py	38
metrics/auroc.py	90
metrics/rce.py	90
ml_logging/__init__.py	1
ml_logging/absl_logging.py	11
ml_logging/torch_logging.py	33
model.py	44
optimizers/__init__.py	1
optimizers/config.py	55
optimizers/optimizer.py	80
projects/__init__.py	1
projects/home/recap/__init__.py	1
projects/home/recap/config.py	37
projects/home/recap/config/local_prod.yaml	476
projects/home/recap/data/__init__.py	1
projects/home/recap/data/config.py	182
projects/home/recap/data/dataset.py	338
projects/home/recap/data/generate_random_data.py	55
projects/home/recap/data/preprocessors.py	142
projects/home/recap/data/tfe_parsing.py	76
projects/home/recap/data/util.py	70
projects/home/recap/embedding/config.py	48
projects/home/recap/main.py	79
projects/home/recap/model/__init__.py	7
projects/home/recap/model/config.py	169
projects/home/recap/model/entrypoint.py	234
projects/home/recap/model/feature_transform.py	89
projects/home/recap/model/mask_net.py	79
projects/home/recap/model/mlp.py	44
projects/home/recap/model/model_and_loss.py	47
projects/home/recap/model/numeric_calibration.py	13
projects/home/recap/optimizer/__init__.py	1
projects/home/recap/optimizer/config.py	23
projects/home/recap/optimizer/optimizer.py	124
projects/twhin/config.py	11
projects/twhin/config/local.yaml	67
projects/twhin/data/config.py	11
projects/twhin/data/data.py	14
projects/twhin/data/edges.py	86
projects/twhin/machines.yaml	12
projects/twhin/metrics.py	14
projects/twhin/models/config.py	41
projects/twhin/models/models.py	108
projects/twhin/optimizer.py	42
projects/twhin/run.py	75
pyproject.toml	17
reader/__init__.py	1
reader/dataset.py	91
reader/dds.py	93
reader/utils.py	64
tools/pq.py	55