File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -186,7 +186,7 @@ def __init__(self,
186
186
self .nets ['net_dec' ] = build_generator (draftnet_decode )
187
187
188
188
self .set_requires_grad ([self .nets ['net_enc' ]], False )
189
- self .set_requires_grad ([self .nets ['net_enc ' ]], False )
189
+ self .set_requires_grad ([self .nets ['net_dec ' ]], False )
190
190
191
191
# define revision-net params
192
192
self .nets ['net_rev' ] = build_generator (revnet_generator )
@@ -322,7 +322,7 @@ def __init__(self,
322
322
self .nets ['net_enc' ] = build_generator (draftnet_encode )
323
323
self .nets ['net_dec' ] = build_generator (draftnet_decode )
324
324
self .set_requires_grad ([self .nets ['net_enc' ]], False )
325
- self .set_requires_grad ([self .nets ['net_enc ' ]], False )
325
+ self .set_requires_grad ([self .nets ['net_dec ' ]], False )
326
326
327
327
# define the first revnet params
328
328
self .nets ['net_rev' ] = build_generator (revnet_generator )
You can’t perform that action at this time.
0 commit comments