IBM/MAX-Skeleton

Dockerfile: Add boilerplate code for pretrained/custom-trained artifact handling

Closed this issue · 0 comments

We need to update the Dockerfile template to add support for model training:

  • Add a build-time parameter that governs whether the Docker image will be built using the pre-trained model assets or custom-trained model assets
  • Add conditional processing code