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 |
|