Skip to content

Bundle Download Large Files #6362

Closed
@ericspod

Description

@ericspod

Is your feature request related to a problem? Please describe.
Many bundles have the large_files.yml file specifying large files (specifically pre-trained weights). The bundles in the GenerativeModels repo are kept separately currently from those in the model zoo so the bundle download program doesn't work quite as expected so users would want to use bundles from a checkout of the repo. These bundles will be moved over but other users may want their own separate hosting for bundles so I feel there is still a need to make it easier to start using a bundle from a checkout.

One step for this is to make it easier to download these large files on the command line or programmatically. This would help people working with checkout code and make it easier to get bundles and their data into Colab and other environments.

I would also suggest changes to the download command we have for bundles to work with a URL pointing to a directory in an arbitrary repo. This is much more complicated now but would allow a user to provide this as the URL and get that whole bundle large files and all. Again this would be helpful in Colab and elsewhere.

Describe the solution you'd like
I would like the code for downloading the large files here to be put into a command in the monai/bundle/scripts.py file so that users can run this and get the files they need to use the bundle.

Describe alternatives you've considered
One has to manually download weights and other files currently which isn't impossible but adds an extra layer of complexity to tell people to look into the yaml file.

Additional context
This was from experience trying to get going quickly with a bundle. We do also need a way of programmatically getting a bundle's network with pre-trained weights like torch.hub.load though this will require some added specification in the bundle.

Metadata

Metadata

Assignees

Labels

BundlesAnything related to bundlesenhancementNew feature or request

Type

No type

Projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions