resnet10

It's done in context of porting `HIL-SERL` paper code (https://hil-serl.github.io/) to `Lerobot` (https://github.com/Lerobot/lerobot). The HF doesn't have ResNet-10 model, which could be pretty usefult for robotics tasks because of it's small size. This model is converted from JAX to PyTorch, and the weights are preserved. This model is a PyTorch port of the original JAX implementation. The conversion maintains the original model's architecture and weights while making it accessible to PyTorch users. The original model is from https://github.com/rail-berkeley/hil-serl/blob/7d17d13560d85abffbd45facec17c4f9189c29c0/serl_launcher/serl_launcher/utils/train_utils.py#L103. - **Original Framework:** JAX - **Target Framework:** PyTorch - **Model Architecture:** ResNet-10 (4-stage ResNet with basic blocks) - **Original Model:** HIL-SERL ResNet-10 - **Total Parameters:** 4,905,792 (~4.9M parameters) - **Hidden Sizes:** [64, 128, 256, 512] - **Input:** 3-channel RGB images (128x128) - **Embedding Size:** 64

번역 준비 중입니다. 원문을 표시합니다.