Update to 150K iteration model checkpoint
Browse files- Updated app.py to load net_g_150000.pth instead of net_g_20000.pth
- Replaced model weights with more recent 150K iteration checkpoint
- Should provide improved super-resolution quality from extended training
🤖 Generated with [Claude Code](https://claude.ai/code)
Co-Authored-By: Claude <noreply@anthropic.com>
- app.py +1 -1
- net_g_20000.pth → net_g_150000.pth +2 -2
app.py
CHANGED
|
@@ -681,7 +681,7 @@ model = HAT(
|
|
| 681 |
)
|
| 682 |
|
| 683 |
# Load the fine-tuned weights
|
| 684 |
-
checkpoint = torch.load('
|
| 685 |
if 'params_ema' in checkpoint:
|
| 686 |
model.load_state_dict(checkpoint['params_ema'])
|
| 687 |
elif 'params' in checkpoint:
|
|
|
|
| 681 |
)
|
| 682 |
|
| 683 |
# Load the fine-tuned weights
|
| 684 |
+
checkpoint = torch.load('net_g_150000.pth', map_location=device)
|
| 685 |
if 'params_ema' in checkpoint:
|
| 686 |
model.load_state_dict(checkpoint['params_ema'])
|
| 687 |
elif 'params' in checkpoint:
|
net_g_20000.pth → net_g_150000.pth
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:54b2eda431411bfecc50c542ec863267ed683128b7aa4163c109776c627bb721
|
| 3 |
+
size 170285859
|