mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-20 19:19:56 +00:00
fix(resnet weights): fix default ResNet weights for audio
This commit is contained in:
@@ -111,7 +111,7 @@ class ACTConfig(PreTrainedConfig):
|
|||||||
replace_final_stride_with_dilation: int = False
|
replace_final_stride_with_dilation: int = False
|
||||||
# Audio backbone.
|
# Audio backbone.
|
||||||
audio_backbone: str = vision_backbone
|
audio_backbone: str = vision_backbone
|
||||||
pretrained_backbone_weights_audio: str | None = ""
|
pretrained_backbone_weights_audio: str | None = None
|
||||||
replace_final_stride_with_dilation_audio: int = False
|
replace_final_stride_with_dilation_audio: int = False
|
||||||
# Transformer layers.
|
# Transformer layers.
|
||||||
pre_norm: bool = False
|
pre_norm: bool = False
|
||||||
|
|||||||
Reference in New Issue
Block a user