refactor: remove redundant .to(device) in StageNet and convert assert to ValueError#966

haoyu-haoyu wants to merge 2 commits intosunlabuiuc:mastersunlabuiuc/PyHealth:masterfrom

haoyu-haoyu:refactor/code-qualityhaoyu-haoyu/PyHealth:refactor/code-qualityCopy head branch name to clipboard