Skip to content

Commit

Permalink
clean up repo (#188)
Browse files Browse the repository at this point in the history
* add mixed precision script

* cleanup

* move scripts
  • Loading branch information
jpata authored Aug 4, 2023
1 parent 8017d54 commit ae5ec59
Show file tree
Hide file tree
Showing 130 changed files with 815 additions and 4,790 deletions.
2 changes: 1 addition & 1 deletion .pre-commit-config.yaml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
default_language_version:
python: python3
exclude: ^(delphes/tev14_pythia8_*.)|^(images/)|^(clic/dumper_hepsim.py)|^(mlpf/pyg/__init__.py)|^(fcc/clicRec_e4h_input.py)|^(fcc/clic_steer.py)|^(fcc/pythia.py)|^(fcc/PandoraSettings)
exclude: ^(scripts/delphes/tev14_pythia8_*.)|^(images)|^(scripts/fcc/clicRec_e4h_input.py)|^(scripts/fcc/clic_steer.py)|^(scripts/fcc/pythia.py)|^(scripts/fcc/PandoraSettings/.*)

repos:
- repo: https://github.com/pre-commit/pre-commit-hooks
Expand Down
5 changes: 0 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,3 @@ MLPF focuses on developing full event reconstruction based on computationally sc
- paper: https://doi.org/10.1140/epjc/s10052-021-09158-w
- code: https://doi.org/10.5281/zenodo.4559587
- dataset: https://doi.org/10.5281/zenodo.4559324

Short instructions with a single test file in [notebooks/delphes-tf-mlpf-quickstart.ipynb](notebooks/delphes-tf-mlpf-quickstart.ipynb).

Long instructions for reproducing the full training from scratch in [README_delphes.md](README_delphes.md).
The plots can be generated using the notebook [delphes/delphes_model_analysis.ipynb](delphes/delphes_model_analysis.ipynb).
60 changes: 60 additions & 0 deletions apptainer/python_base.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
POT
PyYAML
astropy
awkward
awkward0
black
bokeh
boost-histogram
corner
cupy-cuda11x
dask
distributed
docopt
emcee
energyflow
fastjet
fastparquet
flake8
girder-client
hdbscan
healpy
hydra-core
imageio
imageio-ffmpeg
ipyparallel
isort
jupyter
line_profiler
lmfit
lz4
matplotlib
memory_profiler
mpl_scatter_density
mplhep
networkx
notebook
numba
numpy
pandas
papermill
parsl
particle
plotly
pre-commit
pyarrow
pydot
pygraphviz
pymultinest
pynbody
pytest
scikit-learn
scipy
seaborn
tables
tensorboard
tqdm
uproot
vector
xxhash
zenodo_get
17 changes: 17 additions & 0 deletions apptainer/python_tf.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
comet-ml
datasets
keras-flops
keras-tuner
keras_core
onnx
onnxruntime
ray[default]
ray[tune]
tensorboard_plugin_profile
tensorflow-datasets
tensorflow-estimator
tensorflow-model-optimization
tensorflow-text
tf-models-official
tf2onnx
transformers
29 changes: 29 additions & 0 deletions apptainer/tf-2.13.0.singularity
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
Bootstrap: docker

From: tensorflow/tensorflow:2.13.0-gpu-jupyter

%files
python_base.txt /opt/python_base.txt
python_tf.txt /opt/python_tf.txt

%post
apt update -y --fix-missing
apt install -y make cmake parallel gcc g++ gfortran binutils
apt install -y libblas3 libblas-dev liblapack3 liblapack-dev libatlas3-base libatlas-base-dev
apt install -y libtcmalloc-minimal4
apt install -y graphviz graphviz-dev
apt install -y git
apt install -y wget

python3 -m pip install --upgrade pip
python3 -m pip install -r /opt/python_base.txt
python3 -m pip install -r /opt/python_tf.txt
python3 -m pip install hls4ml[profiling]

%environment
export PIP_DEFAULT_TIMEOUT=500
export LC_ALL=""
export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu

%runscript
/bin/bash
20 changes: 0 additions & 20 deletions environment.yml

This file was deleted.

Loading

0 comments on commit ae5ec59

Please sign in to comment.