Skip to content

Commit

Permalink
Update pretrained URLs
Browse files Browse the repository at this point in the history
  • Loading branch information
Gabriel Gouvine authored and Gabriel Gouvine committed Jun 29, 2021
1 parent a084e0c commit e9eaa4c
Show file tree
Hide file tree
Showing 5 changed files with 27 additions and 27 deletions.
4 changes: 2 additions & 2 deletions torchsr/models/carn.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@


urls = {
'carn': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/carn.pt',
'carn_m': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/carn_m.pt',
'carn': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/carn.pt',
'carn_m': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/carn_m.pt',
}


Expand Down
12 changes: 6 additions & 6 deletions torchsr/models/edsr.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,12 +10,12 @@
__all__ = [ 'edsr_baseline', 'edsr', 'edsr_r16f64', 'edsr_r32f256', ]

url = {
'r16f64x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/edsr64_x2.pt',
'r16f64x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/edsr64_x3.pt',
'r16f64x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/edsr64_x4.pt',
'r32f256x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/edsr_x2.pt',
'r32f256x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/edsr_x3.pt',
'r32f256x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/edsr_x4.pt',
'r16f64x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/edsr64_x2.pt',
'r16f64x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/edsr64_x3.pt',
'r16f64x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/edsr64_x4.pt',
'r32f256x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/edsr_x2.pt',
'r32f256x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/edsr_x3.pt',
'r32f256x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/edsr_x4.pt',
}


Expand Down
24 changes: 12 additions & 12 deletions torchsr/models/ninasr.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,18 +7,18 @@
__all__ = [ 'ninasr_b0', 'ninasr_b1', 'ninasr_b2' ]

url = {
'r10f16x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b0_x2.pt',
'r10f16x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b0_x3.pt',
'r10f16x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b0_x4.pt',
'r10f16x8': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b0_x8.pt',
'r26f32x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b1_x2.pt',
'r26f32x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b1_x3.pt',
'r26f32x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b1_x4.pt',
'r26f32x8': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b1_x8.pt',
'r84f56x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b2_x2.pt',
'r84f56x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b2_x3.pt',
'r84f56x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b2_x4.pt',
'r84f56x8': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/ninasr_b2_x8.pt',
'r10f16x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b0_x2.pt',
'r10f16x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b0_x3.pt',
'r10f16x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b0_x4.pt',
'r10f16x8': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b0_x8.pt',
'r26f32x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b1_x2.pt',
'r26f32x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b1_x3.pt',
'r26f32x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b1_x4.pt',
'r26f32x8': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b1_x8.pt',
'r84f56x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b2_x2.pt',
'r84f56x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b2_x3.pt',
'r84f56x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b2_x4.pt',
'r84f56x8': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/ninasr_b2_x8.pt',
}

class AttentionBlock(nn.Module):
Expand Down
8 changes: 4 additions & 4 deletions torchsr/models/rcan.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@
__all__ = [ 'rcan', 'rcan_g10r20f64', ]

url = {
'g10r20f64x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/rcan_x2.pt',
'g10r20f64x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/rcan_x3.pt',
'g10r20f64x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/rcan_x4.pt',
'g10r20f64x8': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/rcan_x8.pt',
'g10r20f64x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/rcan_x2.pt',
'g10r20f64x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/rcan_x3.pt',
'g10r20f64x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/rcan_x4.pt',
'g10r20f64x8': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/rcan_x8.pt',
}

def default_conv(in_channels, out_channels, kernel_size, bias=True):
Expand Down
6 changes: 3 additions & 3 deletions torchsr/models/rdn.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@
__all__ = [ 'rdn', 'rdn_a', 'rdn_b', ]

url = {
'g64go64d16c8x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/rdn_x2.pt',
'g64go64d16c8x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/rdn_x3.pt',
'g64go64d16c8x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.1/rdn_x4.pt',
'g64go64d16c8x2': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/rdn_x2.pt',
'g64go64d16c8x3': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/rdn_x3.pt',
'g64go64d16c8x4': 'https://github.com/Coloquinte/torchSR/releases/download/v1.0.3/rdn_x4.pt',
}

class RDB_Conv(nn.Module):
Expand Down

0 comments on commit e9eaa4c

Please sign in to comment.