diff --git a/README.md b/README.md index 2fcac7c95..dce648d90 100644 --- a/README.md +++ b/README.md @@ -55,7 +55,7 @@ python scripts/run_train.py data/dataset=friends_TBBT-big_living_room # Run with different datasets and config changes python scripts/run_train.py data/dataset=friends_TBBT-big_living_room graph.network.far_plane=14 -# Speed up the dataloading pipeline. +# [Experimental] Speed up the dataloading pipeline by caching DatasetInputs. python scripts/run_data_preprocessor.py # Then, specify using the cache. python scripts/run_train.py ++data.dataset.use_cache=true diff --git a/environment/requirements.txt b/environment/requirements.txt index 1522648c7..d690d239f 100644 --- a/environment/requirements.txt +++ b/environment/requirements.txt @@ -13,10 +13,8 @@ protobuf==3.20.0 pylint==2.13.4 pytest tensorboard==2.9.0 ---extra-index-url https://download.pytorch.org/whl/cu113 torch==1.11.0 torchmetrics[image]==0.8.2 torchtyping ---extra-index-url https://download.pytorch.org/whl/cu113 torchvision==0.12.0 tqdm \ No newline at end of file