mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Disable efficient_builder debug flag

Browse Source
This commit is contained in:
Ross Wightman 2024-06-06 21:50:27 -07:00
parent ad026e6e33
commit 7ccb10ebff
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/_efficientnet_builder.py
View File

@ -25,7 +25,7 @@ __all__ = ["EfficientNetBuilder", "decode_arch_def", "efficientnet_init_weights"
_logger = logging.getLogger(__name__)
_DEBUG_BUILDER = True
_DEBUG_BUILDER = False
# Defaults used for Google/Tensorflow training of mobile networks /w RMSprop as per
# papers and TF reference implementations. PT momentum equiv for TF decay is (1 - TF decay)