Model description
This repo contains the trained model for Style transfer using vgg16 as the backbone.
Full credits go to Nhu Hoang
Motivation: Style transfer is an interesting task with an amazing outcome.
Training and evaluation data
Training hyperparameters
The following hyperparameters were used during training:
| Hyperparameters | Value |
|---|---|
| name | Adam |
| learning_rate | 3e-5 |
| training_precision | float16 |
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The HF Inference API does not support image-to-image models for fastai library.