Update README.md
Browse files
README.md
CHANGED
|
@@ -27,6 +27,8 @@ Training was conducted using FP16 mixed-precision and DeepSpeed Zero2 scheme. Th
|
|
| 27 |
was kept frozen during the training. We used the [diffusers/ShotDEAD-v0](https://huggingface.co/datasets/diffusers/ShotDEAD-v0)
|
| 28 |
dataset for conducting training.
|
| 29 |
|
|
|
|
|
|
|
| 30 |
## Inference
|
| 31 |
|
| 32 |
```py
|
|
|
|
| 27 |
was kept frozen during the training. We used the [diffusers/ShotDEAD-v0](https://huggingface.co/datasets/diffusers/ShotDEAD-v0)
|
| 28 |
dataset for conducting training.
|
| 29 |
|
| 30 |
+
Training is available [here](https://github.com/huggingface/movie-shot-categorizer).
|
| 31 |
+
|
| 32 |
## Inference
|
| 33 |
|
| 34 |
```py
|