gitextract_800cdojr/

├── .github/
│   └── workflows/
│       └── main.yml
├── .gitignore
├── .pre-commit-config.yaml
├── COPYING
├── LICENSE.torchrec
├── README.md
├── common/
│   ├── __init__.py
│   ├── batch.py
│   ├── checkpointing/
│   │   ├── __init__.py
│   │   └── snapshot.py
│   ├── device.py
│   ├── filesystem/
│   │   ├── __init__.py
│   │   ├── test_infer_fs.py
│   │   └── util.py
│   ├── log_weights.py
│   ├── modules/
│   │   └── embedding/
│   │       ├── config.py
│   │       └── embedding.py
│   ├── run_training.py
│   ├── test_device.py
│   ├── testing_utils.py
│   ├── utils.py
│   └── wandb.py
├── core/
│   ├── __init__.py
│   ├── config/
│   │   ├── __init__.py
│   │   ├── base_config.py
│   │   ├── base_config_test.py
│   │   ├── config_load.py
│   │   ├── test_config_load.py
│   │   └── training.py
│   ├── custom_training_loop.py
│   ├── debug_training_loop.py
│   ├── loss_type.py
│   ├── losses.py
│   ├── metric_mixin.py
│   ├── metrics.py
│   ├── test_metrics.py
│   ├── test_train_pipeline.py
│   └── train_pipeline.py
├── images/
│   ├── init_venv.sh
│   └── requirements.txt
├── machines/
│   ├── environment.py
│   ├── get_env.py
│   ├── is_venv.py
│   └── list_ops.py
├── metrics/
│   ├── __init__.py
│   ├── aggregation.py
│   ├── auroc.py
│   └── rce.py
├── ml_logging/
│   ├── __init__.py
│   ├── absl_logging.py
│   ├── test_torch_logging.py
│   └── torch_logging.py
├── model.py
├── optimizers/
│   ├── __init__.py
│   ├── config.py
│   └── optimizer.py
├── projects/
│   ├── __init__.py
│   ├── home/
│   │   └── recap/
│   │       ├── FEATURES.md
│   │       ├── README.md
│   │       ├── __init__.py
│   │       ├── config/
│   │       │   ├── home_recap_2022/
│   │       │   │   └── segdense.json
│   │       │   └── local_prod.yaml
│   │       ├── config.py
│   │       ├── data/
│   │       │   ├── __init__.py
│   │       │   ├── config.py
│   │       │   ├── dataset.py
│   │       │   ├── generate_random_data.py
│   │       │   ├── preprocessors.py
│   │       │   ├── tfe_parsing.py
│   │       │   └── util.py
│   │       ├── embedding/
│   │       │   └── config.py
│   │       ├── main.py
│   │       ├── model/
│   │       │   ├── __init__.py
│   │       │   ├── config.py
│   │       │   ├── entrypoint.py
│   │       │   ├── feature_transform.py
│   │       │   ├── mask_net.py
│   │       │   ├── mlp.py
│   │       │   ├── model_and_loss.py
│   │       │   └── numeric_calibration.py
│   │       ├── optimizer/
│   │       │   ├── __init__.py
│   │       │   ├── config.py
│   │       │   └── optimizer.py
│   │       └── script/
│   │           ├── create_random_data.sh
│   │           └── run_local.sh
│   └── twhin/
│       ├── README.md
│       ├── config/
│       │   └── local.yaml
│       ├── config.py
│       ├── data/
│       │   ├── config.py
│       │   ├── data.py
│       │   ├── edges.py
│       │   ├── test_data.py
│       │   └── test_edges.py
│       ├── machines.yaml
│       ├── metrics.py
│       ├── models/
│       │   ├── config.py
│       │   ├── models.py
│       │   └── test_models.py
│       ├── optimizer.py
│       ├── run.py
│       ├── scripts/
│       │   ├── docker_run.sh
│       │   └── run_in_docker.sh
│       └── test_optimizer.py
├── pyproject.toml
├── reader/
│   ├── __init__.py
│   ├── dataset.py
│   ├── dds.py
│   ├── test_dataset.py
│   ├── test_utils.py
│   └── utils.py
└── tools/
    └── pq.py