Make model export directory easier

[1vn]

Description

• Create a version directory for them if it's not one already

Bonus:

• Recursively walk the directory and extract the sub directory that matches the export structure. Then people can just point to their checkpoint dir.