Want to Stop Rebuilding "Expensive" Parts of your Flow? Explicit Builds are the Answer!READ MORE

Image classification using pytorch

Gayatri
Level 1
Image classification using pytorch

Hi ..can we use hugging face pytorch and transformer model in dataiku and retrain the existing model of hugging face in dataiku on different image dataset such as retina images.??

0 Kudos
1 Reply
JakeA
Dataiker
Dataiker

If I am understanding your question correctly, you are asking if you may use both pytorch and hugging face. To do this you will need to configure your code environment to make sure you have both of these packages. From there you will most likely need to write some custom code in order to achieve this, but it seems like a possible task within the DSS.

0 Kudos