File tree Expand file tree Collapse file tree 1 file changed +10
-5
lines changed Expand file tree Collapse file tree 1 file changed +10
-5
lines changed Original file line number Diff line number Diff line change @@ -175,11 +175,16 @@ Our BiRefNet has achieved SOTA on many similar HR tasks:
175
175
> The devices used in the below table differ from those in the original paper (the standard). So, it's only for reference.
176
176
177
177
| Runtime | * FP32* | * FP16* |
178
- | :----: | :----: | :----: |
179
- | GPUMem | 4.76GB | 3.45GB |
180
- | A100 | 86.8ms | 69.4ms |
181
- | 4090 | 95.8ms | 57.7ms |
182
- | V100 | 384ms | 152ms |
178
+ | :-----: | :----: | :----: |
179
+ | A100 | 86.8ms | 69.4ms |
180
+ | 4090 | 95.8ms | 57.7ms |
181
+ | V100 | 384ms | 152ms |
182
+
183
+ | GPU Memory | * FP32* | * FP16* |
184
+ | :--------: | :----: | :----: |
185
+ | Inference | 4.76GB | 3.45GB |
186
+ | Training (\# GPU=1, batch\_ size=2, compile=False+PyTorch=2.5.1) | 36.3GB | 30.4GB |
187
+ | Training (\# GPU=1, batch\_ size=2, compile=True+PyTorch=2.5.1) | 35.9GB | ** 24.9GB** |
183
188
184
189
</details >
185
190
You can’t perform that action at this time.
0 commit comments