Published August 20, 2020 | Version 1.0.0
Software Open

Keras model and weights for GarNet-on-FPGA

  • 1. The University of Tokyo

Description

Source code and input weights data for GarNet-on-FPGA

This repository contains the Keras layer and model files for the simultaneous regression and classification task described in arXiv:2008.03601. Dataset in 10.5281/zenodo.3888910 can be fed into the training script (train.py) to reproduce the results in the paper. The GarNet layer file (garnet.py) is a partial copy from the original at commit 6d1127d.

The hls4ml configuration and weights files for the "continuous" and "quantized" models in the paper (Vmax = 128) are also packaged together in the tarball hls4ml_inputs.tar.gz. HLS workspaces can be created from the input files with hls4ml version > v0.3.0 (commit e86c103 and later)

Files

Files (50.7 kB)

Name Size Download all
md5:462874125a8a38319cda090d24f92058
12.8 kB Download
md5:a192347100d343fc4cd310f21ca0b3a3
904 Bytes Download
md5:1f21c06efcc14c2ee740b2d52de3873d
32.8 kB Download
md5:3855b142e244d64d802a8e8110fd0d3b
1.1 kB Download
md5:b14fcd3dae9ea231048ae212c74ff03a
3.1 kB Download

Additional details

Related works

Is cited by
Preprint: arXiv:2008.03601 (arXiv)
Is derived from
Software: https://github.com/jkiesele/caloGraphNN (URL)
References
Dataset: 10.5281/zenodo.3888910 (DOI)

Funding

mPP – machine learning for Particle Physics 772369
European Commission