Update README.md
Browse files
README.md
CHANGED
|
@@ -241,7 +241,7 @@ DataParallel(
|
|
| 241 |
)
|
| 242 |
)
|
| 243 |
```
|
| 244 |
-
|
| 245 |
**Training hardware:** Each of the models were trained on 2 x T4 GPUs (multi-GPU training). For this reason, linear attention modules were implemented as ring (distributed) attention during training.
|
| 246 |
**Total Training Compute Throughput:** 4.13 TFLOPS
|
| 247 |
**Total Logged Training Time:** ~210 hours (total time split across four models including overhead)
|
|
@@ -253,33 +253,22 @@ DataParallel(
|
|
| 253 |
- v1_3: 355 MB
|
| 254 |
- v2_2: 11 MB
|
| 255 |
|
| 256 |
-
## Evaluation
|
| 257 |
-
|
| 258 |
-
<!-- This section describes the evaluation protocols and provides the results. -->
|
| 259 |
-
|
| 260 |
-
### Testing Data, Factors & Metrics
|
| 261 |
-
|
| 262 |
-
#### Testing Data
|
| 263 |
-
|
| 264 |
-
<!-- This should link to a Dataset Card if possible. -->
|
| 265 |
-
|
| 266 |
-
[More Information Needed]
|
| 267 |
-
|
| 268 |
-
#### Factors
|
| 269 |
|
| 270 |
-
|
| 271 |
|
| 272 |
-
|
| 273 |
|
| 274 |
-
|
| 275 |
|
| 276 |
-
|
| 277 |
|
| 278 |
-
|
|
|
|
| 279 |
|
| 280 |
### Results
|
| 281 |
|
| 282 |
-
|
| 283 |
|
| 284 |
#### Summary
|
| 285 |
|
|
|
|
| 241 |
)
|
| 242 |
)
|
| 243 |
```
|
| 244 |
+
|
| 245 |
**Training hardware:** Each of the models were trained on 2 x T4 GPUs (multi-GPU training). For this reason, linear attention modules were implemented as ring (distributed) attention during training.
|
| 246 |
**Total Training Compute Throughput:** 4.13 TFLOPS
|
| 247 |
**Total Logged Training Time:** ~210 hours (total time split across four models including overhead)
|
|
|
|
| 253 |
- v1_3: 355 MB
|
| 254 |
- v2_2: 11 MB
|
| 255 |
|
| 256 |
+
## Evaluation Data, Metrics & Results
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 257 |
|
| 258 |
+
This section covers information on how the model was evaluated at each stage.
|
| 259 |
|
| 260 |
+
### Evaluation Data
|
| 261 |
|
| 262 |
+
Evaluation was performed on real-time footage captured from Grand Theft Auto V, Cyberpunk 2077 and WatchDogs 2.
|
| 263 |
|
| 264 |
+
### Metrics
|
| 265 |
|
| 266 |
+
- PSNR (Peak Signal-to-Noise Ratio)
|
| 267 |
+
- Combined loss (L1 loss + Total Variation loss)
|
| 268 |
|
| 269 |
### Results
|
| 270 |
|
| 271 |
+
|
| 272 |
|
| 273 |
#### Summary
|
| 274 |
|