luni/cvpr_2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pyramid model fix

Browse Source
This commit is contained in:
luni
2023-05-22 23:20:15 +02:00
parent 08bf60a76a
commit 7d17b1a222
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/pyramid_model.py
+1 -1
View File
@@ -154,7 +154,7 @@ class Pyramid(nn.Module):
for param in self.body_enc.parameters(): param.requires_grad = False for param in self.body_enc.parameters(): param.requires_grad = False
for param in self.curv_enc.parameters(): param.requires_grad = False for param in self.curv_enc.parameters(): param.requires_grad = False
self.levels = [PyramidLevel({ **config['pyramid']['levels'][i] }) for i in range(len(config['pyramid']['levels']))] self.levels = nn.ModuleList([PyramidLevel({ **config['pyramid']['levels'][i] }) for i in range(len(config['pyramid']['levels']))])
# if len(config['pyramid']['resolutions']) == 2: # if len(config['pyramid']['resolutions']) == 2: