Artera (CA) interview question

1) Implement class-balanced train-val split 2) Debug/generalize Pytorch model