Skip to content

Commit

Permalink
Revert "fix timm conversion script"
Browse files Browse the repository at this point in the history
This reverts commit 439f0a5.
  • Loading branch information
sachinprasadhs committed Sep 5, 2024
1 parent 439f0a5 commit dbaeabd
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions keras_nlp/src/utils/timm/convert_resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,8 +56,6 @@ def convert_backbone_config(timm_config):
stackwise_num_strides=[1, 2, 2, 2],
block_type=block_type,
use_pre_activation=use_pre_activation,
input_conv_filters=[64],
input_conv_kernel_sizes=[7],
)


Expand Down Expand Up @@ -102,10 +100,10 @@ def port_batch_normalization(keras_layer_name, hf_weight_prefix):
for stack_index in range(num_stacks):
for block_idx in range(backbone.stackwise_num_blocks[stack_index]):
if version == "v1":
keras_name = f"stack{stack_index}_block{block_idx}"
keras_name = f"v1_stack{stack_index}_block{block_idx}"
hf_name = f"layer{stack_index+1}.{block_idx}"
else:
keras_name = f"stack{stack_index}_block{block_idx}"
keras_name = f"v2_stack{stack_index}_block{block_idx}"
hf_name = f"stages.{stack_index}.blocks.{block_idx}"

if version == "v1":
Expand Down

0 comments on commit dbaeabd

Please sign in to comment.