why the file in ckpts folder is h5 file

limhoyeon / ToothGroupNetwork

3D Dental surface segmentation with Tooth Group Network

173 stars 41 forks source link

Closed cookie-csooo closed 1 year ago

cookie-csooo commented 1 year ago

why not pth file

limhoyeon commented 1 year ago

There is no specific reason. The extension '.pth' is just a file extension and there will be no issues loading ".h5" file in the PyTorch code.

cookie-csooo commented 1 year ago

thanks for your answer and share