peace / boxgraph /README.md
davidbraun's picture
Release PEACE BoxGraph and T5 models
484fe3c verified
|
Raw History Blame Contribute Delete
816 Bytes
metadata
library_name: peace
license: cc-by-nc-4.0
tags:
  - audio
  - audio-effects
  - audio-retrieval
  - code-retrieval
  - faust
  - jax
  - flax

PEACE: AFx-Rep with BoxGraph

The BoxGraph code encoder runs four weight-shared message-passing layers of width 1024 over the Faust Block Diagram Algebra graph of a program, and mean-pools its seq and effect nodes. Chains deeper than training run one layer per graph level. Embeddings depend only on the parsed program, not on how it is written. This is run 57910cd3 at step 36000.

Audio→Code R@1 R@10 Code→Audio R@1 R@10
48.5% 77.7% 50.8% 77.5%
from peace import PEACE

model = PEACE.from_pretrained("davidbraun/peace", subfolder="boxgraph")

See the repository README for the full model card.