NVIDIA-AI-IOT / nvidia-tao

Other
82 stars 11 forks source link

pre-trained models #8

Open tugbakara opened 1 year ago

tugbakara commented 1 year ago

Is it possible to use TAO with my custom dataset and my custom pre-trained model instead og using models from NGC? Is any way exists which files of my pretrained model do I need to put in this linkIn colab NGC pre-trained modelsb are installed to EXPERIMENT_DIR and specs are in different directory, when I install example pretrained it gives hdf5 extension file but I have .weights so how can I achieve them? I really confused.