Felix-ML

A machine learning approach to diffraction patterns.
Check out the webpage to see documentation and research!

Quick Start

Read "Installation" from

Train model with default configuration

# train on CPU
python src/train.py trainer=cpu

# train on GPU
python src/train.py trainer=gpu

Train model with chosen experiment configuration from configs/experiment/

python src/train.py experiment=experiment_name.yaml

You can override any parameter from command line like this

python src/train.py trainer.max_epochs=20 data.batch_size=64

Create a sweep over hyperparameters

python train.py -m data.batch_size=32,64,128 model.lr=0.001,0.0005

Name		Name	Last commit message	Last commit date
Latest commit History 117 Commits
configs		configs
data		data
docs		docs
logs		logs
notebooks		notebooks
scripts		scripts
src		src
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
.project-root		.project-root
Makefile		Makefile
README.md		README.md
environment.yaml		environment.yaml
pyproject.toml		pyproject.toml
requirements.txt		requirements.txt