naksyu/uzr_3brains

https://github.com/10kseason/uzr 참조바랍니다.

python -m uzr.train_meta_3brains --device cuda --steps 15000 --z_slow_lang_dim 96 --z_slow_logic_dim 96 --z_bridge_dim 64 --lam_lang 5e-4 --lam_logic 5e-4 --lam_bridge 3e-4 --inner_steps 6 --inner_eta 0.425 --identity "루리아" --save uzr_3brains_ckpt.pt --self_eval on --abstain --amp --save_every 200

로 트레인되었습니다.