Skip to content

Latest commit

 

History

History
46 lines (26 loc) · 878 Bytes

README.md

File metadata and controls

46 lines (26 loc) · 878 Bytes
Raw

ReasonNet

generate data

cd data_gen
cargo run --release -- --min 3 --max 5 --size 20000000 --seed 32 --save-path data-20m-add --chunk-size=500000 --operators=+

mv data-20m-add ../datasets/.


cargo run --release -- --min 3 --max 5 --size 20000000 --seed 32 --save-path data-20m-all --chunk-size=500000 
mv data-20m-all ../datasets/.

run the training

normal

python reason_net/run.py --config-path configs --config-name all-14m.yaml

reason middle

python reason_net/run.py --config-path configs --config-name all-14m.yaml reason_mode=true

reason left

python reason_net/run.py --config-path configs --config-name all-14m.yaml reason_mode=true +data.reason.reason_token_pos="left"

With small model for test

python reason_net/run.py --config-path configs --config-name all-14m.yaml module/model=910K