Skip to content

Commit d62d1ae

Browse files
authored
Update train_semi.py
Added "net" to cfg for making this train_semi.py detect
1 parent 5784b29 commit d62d1ae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train_semi.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def main():
8585
else:
8686
modules_head = [model.decoder]
8787

88-
if cfg.get("sync_bn", True):
88+
if cfg["net"].get("sync_bn", True):
8989
model = torch.nn.SyncBatchNorm.convert_sync_batchnorm(model)
9090

9191
model.cuda()

0 commit comments

Comments
 (0)