optim.Adam(list(net1.parameters()) + list(net2.parameters()),
lr=cfg.learning_rate_apt,
betas=(cfg.beta1, cfg.beta2))


optim.Adam(itertools(net1.parameters(), net2.parameters()),
lr=cfg.learning_rate_apt,
betas=(cfg.beta1, cfg.beta2))