liav
/
lut_reproduce
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
main
Vladimir Protsenko 8 months ago
parent 7d077571cf
commit 17bb7aa882
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File

1
src/models/__init__.py
Unescape Escape View File

@ -20,6 +20,7 @@ AVAILABLE_MODELS = {
'SDYNetx1': sdynet.SDYNetx1, 'SDYNetx1': sdynet.SDYNetx1,
'RCNetx2Unlutable': rcnet.RCNetx2Unlutable, 'RCNetx2Unlutable': rcnet.RCNetx2Unlutable,
'RCNetx2CenteredUnlutable': rcnet.RCNetx2CenteredUnlutable, 'RCNetx2CenteredUnlutable': rcnet.RCNetx2CenteredUnlutable,
'RCNetCentered_7x7v2': rcnet.RCNetCentered_7x7v2
} }
def SaveCheckpoint(model, path): def SaveCheckpoint(model, path):

Write Preview
Loading…
Cancel
Save