BorisEm Claude commited on
Commit
c47527a
·
1 Parent(s): 67aaece

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>

Files changed (2) hide show
  1. app.py +1 -1
  2. 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('net_g_20000.pth', map_location=device)
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:94d74fbc11a23bec569dd31b9f10b1b1033fd839cadabbef155d2beab3a3ffeb
3
- size 170284129
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54b2eda431411bfecc50c542ec863267ed683128b7aa4163c109776c627bb721
3
+ size 170285859