SORL:
python3 train_style_lora_m2.py ../atari_data/space_invaders game models_atari_head/dqn_models/space_dqn_onlydqn/ --epochs 30 --workers 16 --framestack 2 --l2 0.00001 --save-freq 1 --merge --atari-head

Off-RLPMM:
python3 train_style_lora_01.py ../atari_data/space_invaders game models_atari_head/dqn_models/space_dqn_onlydqn/ --epochs 30 --workers 16 --framestack 2 --l2 0.00001 --save-freq 1 --merge --atari-head

InfoGAIL:
python3 train_style_lora_infogail.py ../atari_data/space_invaders game models_atari_head/dqn_models/space_dqn_onlydqn/ --epochs 30 --workers 16 --framestack 2 --l2 0.00001 --save-freq 1 --merge --atari-head