Can you update your train.py code?

[jiku100]

When I'm training model by using your train.py, There are many errors in your code. For example, In your codes, dataloaders.la_heart, VNet etc. are not in your Codes. So, Can you update your train.py code?

[eraserNut]

Thank you for your advice, I will update it.