Skip to content

Fix convert_sync_batchnorm for timm#1674

Open
emailweixu wants to merge 1 commit intopytorchfrom
PR_timm_convert_sync_batchnorm
Open

Fix convert_sync_batchnorm for timm#1674
emailweixu wants to merge 1 commit intopytorchfrom
PR_timm_convert_sync_batchnorm

Conversation

@emailweixu
Copy link
Contributor

timm (PyTorch Image Models) models may contain BatchNormAct2d, which is not supported by torch.nn.SyncBatchNorm. Need to use timm's convert_sync_batchnorm if timm is used.

timm (PyTorch Image Models) models may contain BatchNormAct2d, which
is not supported by torch.nn.SyncBatchNorm. Need to use timm's
convert_sync_batchnorm if timm is used.
@emailweixu emailweixu requested a review from hnyu July 23, 2024 23:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant