search

GeekyYouthsInfo / Deep-Learning-Cataract-Detection

Academic: Deep learning-based mobile app to accurately identify the presence and severity of cataracts, enabling early diagnosis and treatment.
0 stars 1 forks source link

Modifie the train notebook #15

Closed jona42-ui closed 1 month ago

jona42-ui commented 2 months ago

Added additions to the train notebook with over 1k+ train and validation datasets for 30 epochs The model is save as final_model.pth on my local machine with a size of 226.7MBs