search

NVIDIA / DIGITS

Deep Learning GPU Training System
https://developer.nvidia.com/digits
BSD 3-Clause "New" or "Revised" License
4.12k stars 1.38k forks source link

UNET not in model store / wrong url (or missing environment variable) in docker image 6.0 for model store #1823

Open horvathj opened 7 years ago

horvathj commented 7 years ago

Hi,

The NVIDIA digits website: "Pre-trained UNET model added to the DIGITS model store for image segmentation of medical images" https://developer.nvidia.com/digits I can not see this model in the store. Is there a problem in my installation, or you didnt released it yet? Im using NVIDIA Digits Docker 6 RC 1

JakaCikac commented 7 years ago

I only get 5 models in the Model Store -> LeNet, AlexNet, VGG16, GoogLeNet, DetectNet.

On some screenshots many more were shown..

horvathj commented 7 years ago

@JakaCikac https://github.com/NVIDIA/DIGITS/issues/1553

ethantang95 commented 7 years ago

From digits 6 rc to digits 6 full release, the model store link target has changed. Try upgrading to digits 6

horvathj commented 7 years ago

I have downloaded 13 days ago the digits6 (final) docker image. In: /usr/local/lib/python2.7/dist-packages/digits/config file the URL is this:

if 'DIGITS_MODEL_STORE_URL' in os.environ: url_list = os.environ['DIGITS_MODEL_STORE_URL'] else: url_list = "http://developer.download.nvidia.com/compute/machine-learning/modelstore/5.0"

after changing 5.0 to 6.0 and restarting the docker image now I can see the other models!

thank you for the information.

JakaCikac commented 7 years ago

@horvathj , I am quite new at Docker, what command should be run to be able to edit that file inside the docker image?

@ethantang95 Will the link be updated in the released docker image?

augustoicaro commented 7 years ago

@horvathj I set the environment variable in my local installation like you said, DIGITS_MODEL_STORE_URL=http://developer.download.nvidia.com/compute/machine-learning/modelstore/6.0, but the UNET still missing. I downloaded the source few days ago.

Any news about that?

augustoicaro commented 6 years ago

@ethantang95 Now I understand what you answered. For some reason the digits releases pages only show the 6.0.rc2 and not the final version. So I cloned the master branch, but in master branch the url in digits/config/store_option.py still outdated. So I downloaded the branch digits-6.0 and now everything works fine!