segmentation tutorial fix

sony / model_optimization

Model Compression Toolkit (MCT) is an open source project for neural network model optimization under efficient, constrained hardware. This project provides researchers, developers, and engineers advanced quantization and compression tools for deploying state-of-the-art neural networks.

Apache License 2.0

331 stars 53 forks source link

Pull Request Description:

tutorial fails when running with google golab gpu. fix supports use of gpu.

Checklist before requesting a review:

[x] I set the appropriate labels on the pull request.
[x] I have added/updated the release note draft (if necessary).
[x] I have updated the documentation to reflect my changes (if necessary).
[x] All function and files are well documented.
[x] All function and classes have type hints.
[x] There is a licenses in all file.
[x] The function and variable names are informative.
[x] I have checked for code duplications.
[x] I have added new unittest (if necessary).

sony / model_optimization

segmentation tutorial fix #1228

Pull Request Description:

Checklist before requesting a review: