The implementation of paper "domain generalization on medical imaging classification using episodic training with task augmentation"