python train.py --lr 3e-3 --lr-pool 3e-2 --task closed_form --gtype 0 --seed 0
python train.py --lr 3e-3 --lr-pool 3e-2 --task map_mu --gtype 0 --seed 0
python train.py --lr 3e-3 --lr-pool 1e-2 --task map_sigma --gtype 0 --seed 0
