Training Capabilities? about nuke-ml-server HOT 4 OPEN

Great question!

This is high on our list of features, and we're hoping to have a release of something this week.

Re: passing data back to the model, yes, it's possible. We haven't explored this a lot yet but it's where we expect all of the interesting work to be.

Re: checkpoints, the TensorFlow training example we'll be adding will include a 'checkpoints' folder.

Re: data rate, from our tests it seems training in Docker is roughly on par with training outside Docker.

from nuke-ml-server.

Hi @samhodge,

We have just added training capabilities through the trainingTemplateTF model. You can find instructions on how to train and infer models using this template here.

It basically enables image-to-image training of an encoder-decoder model as follows:

1. Fill in your groundtruth and input images in the data folder,
2. Launch the training inside your Docker container,
3. Infer using your trained model in Nuke through the nuke-ML-server.

from nuke-ml-server.

Is there scope to train something beyond a simple encoder decoder?

from nuke-ml-server.

At the moment, the multi-scale encoder-decoder is the only supported model. We are planning on adding new model building blocks to the template along the way.

In the meantime, you are always free to modify or add your own models inside the model_builder.py file!

from nuke-ml-server.