mit-han-lab / tiny-training

On-Device Training Under 256KB Memory [NeurIPS'22]
http://tinytraining.mit.edu
MIT License
429 stars 59 forks source link

How to get the files in the "ir_zoos" directory which used in mcu_ir_gen.py ? #8

Closed VisionShilin closed 1 year ago

VisionShilin commented 1 year ago

How to get the files in the "ir_zoos" directory which used in mcu_ir_gen.py ?

Lyken17 commented 1 year ago

This is a output directory rather than model checkpoints as you can find in https://github.com/mit-han-lab/tiny-training/blob/main/compilation/mcu_ir_gen.py#L88

Once you run the scripts, corresponding IRs will be generated under ir_zoos

VisionShilin commented 1 year ago

Thanks.