rename files and add project scripts (#566)

* scripts

* b

* fix file
This commit is contained in:
Ethan Weber authored and GitHub committed 2022-09-27 15:02:55 -07:00
1 parent 8af3a7a385
commit cf34ef4bd0
26 files changed
+384 -355

No files matched your search

+1 -1
View File
@@ -59,7 +59,7 @@ for dataset in ${DATASETS[@]}; do
export CUDA_VISIBLE_DEVICES=${GPU_IDX[$idx]}
base_config_name=${config_name/"graph_"/""}
config_path="${output_dir}/blender_${dataset}_${month}-${date}-${year}/${base_config_name}/${year}-${month}-${date}_${seconds}/nerfactory_models/"
python scripts/run_eval.py compute-psnr \
python scripts/eval.py compute-psnr \
--load-config=${config_path} \
--output-path=${output_dir}/${base_config_name}/blender_${dataset}_${month}-${date}-${year}_${seconds}.json &
echo "Launched ${checkpoint_dir} on gpu ${GPU_IDX[$idx]}"
+1 -1
View File
@@ -52,7 +52,7 @@ len=${#GPU_IDX[@]}
for dataset in ${DATASETS[@]}; do
export CUDA_VISIBLE_DEVICES=${GPU_IDX[$idx]}
python scripts/run_train.py \
python scripts/train.py \
${config_name} \
--logging.writer.2.no-enable \
--pipeline.datamanager.train-dataset.data-directory=data/blender/${dataset} \
+8 -3
View File
@@ -237,14 +237,19 @@ def main(
CONSOLE.print()
CONSOLE.print("Notes:")
CONSOLE.print(
":warning: In bash, completions will trigger via [dim white]./scripts/run_train.py <TAB>[/dim white], but"
" not [dim white]python ./scripts/run_train.py <TAB>[/dim white]."
":warning: In bash, completions will trigger via [dim white]./scripts/train.py <TAB>[/dim white], but"
" not [dim white]python ./scripts/train.py <TAB>[/dim white]."
)
CONSOLE.print(
":warning: Completions are in an experimental state. If you run into any issues, please file an issue!"
)
if __name__ == "__main__":
def entrypoint():
"""Entrypoint for use with pyproject scripts."""
dcargs.extras.set_accent_color("bright_yellow")
dcargs.cli(main, description=__doc__)
if __name__ == "__main__":
entrypoint()
+7 -2
View File
@@ -1,6 +1,6 @@
#!/usr/bin/env python
"""
run_eval.py
eval.py
"""
from __future__ import annotations
@@ -197,7 +197,8 @@ class RenderTrajectory:
)
if __name__ == "__main__":
def entrypoint():
"""Entrypoint for use with pyproject scripts."""
# A Union over dataclass types will create a subcommand for each type.
#
# TODO: it would make sense to split this script up into separate scripts.
@@ -208,3 +209,7 @@ if __name__ == "__main__":
# checkpoint when resuming a training run.
dcargs.extras.set_accent_color("bright_yellow")
dcargs.cli(Union[ComputePSNR, RenderTrajectory]).main()
if __name__ == "__main__":
entrypoint()
+6 -1
View File
@@ -482,6 +482,11 @@ def main(
CONSOLE.rule()
if __name__ == "__main__":
def entrypoint():
"""Entrypoint for use with pyproject scripts."""
dcargs.extras.set_accent_color("bright_yellow")
dcargs.cli(main)
if __name__ == "__main__":
entrypoint()
+7 -2
View File
@@ -1,6 +1,6 @@
#!/usr/bin/env python
"""
run_train_nerf.py
train.py
"""
from __future__ import annotations
@@ -204,7 +204,12 @@ def main(config: cfg.Config) -> None:
)
if __name__ == "__main__":
def entrypoint():
"""Entrypoint for use with pyproject scripts."""
# Choose a base configuration and override values.
dcargs.extras.set_accent_color("bright_yellow")
main(dcargs.cli(AnnotatedBaseConfigUnion))
if __name__ == "__main__":
entrypoint()