search

yilundu / irem_code_release

ICML 2022: Learning Iterative Reasoning through Energy Minimization
https://energy-based-model.github.io/iterative-reasoning-as-energy-minimization/
MIT License
42 stars 6 forks source link

Implementing command line lr #3

Open mcleish7 opened 1 year ago

mcleish7 commented 1 year ago

Taking the command line input for lr as it was previously hardcoded so the argument did not change its value. Also removing some unneeded imports.