flare-2022

Train using CLI

Create a config in configs directory similar to ones already there.

source .venv/bin/activate

python trainer_cli.py fit --config configs/version1.yaml

Name		Name	Last commit message	Last commit date
Latest commit History 40 Commits
.vscode		.vscode
configs		configs
custom_transforms		custom_transforms
datamodules		datamodules
models		models
notebooks		notebooks
visualizations		visualizations
.dockerignore		.dockerignore
.gitignore		.gitignore
3d_img_to_gif.py		3d_img_to_gif.py
Dockerfile		Dockerfile
LICENSE		LICENSE
README.md		README.md
__init__.py		__init__.py
c2f_trainer.py		c2f_trainer.py
calculate_params.py		calculate_params.py
calculate_strides.py		calculate_strides.py
checkpoint_converter.py		checkpoint_converter.py
coarse_checkpoint_converter.py		coarse_checkpoint_converter.py
coarse_flare_model.ts		coarse_flare_model.ts
create_torchscript.py		create_torchscript.py
custom_loss.py		custom_loss.py
data_stats.py		data_stats.py
docker_requirements.txt		docker_requirements.txt
fine_flare_model.ts		fine_flare_model.ts
predict.py		predict.py
predict.sh		predict.sh
predict_c2f.py		predict_c2f.py
predict_c2f_scratch.py		predict_c2f_scratch.py
predict_scratch.py		predict_scratch.py
requirements-dev.txt		requirements-dev.txt
requirements.txt		requirements.txt
saver.py		saver.py
setup.cfg		setup.cfg
test_shape.py		test_shape.py
trainer.py		trainer.py
trainer_cli.py		trainer_cli.py
transforms.py		transforms.py