OliverRensu / ARM

This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision
53 stars 1 forks source link

Why does ARM/main_pretrain.py use the VisionMamba model, while ARM/Finetuning/main_finetune.py uses the ARM model? Wouldn't this cause issues? #7

Closed WenjunHuang94 closed 2 weeks ago

WenjunHuang94 commented 3 weeks ago

Why does ARM/main_pretrain.py use the VisionMamba model, while ARM/Finetuning/main_finetune.py uses the ARM model? Wouldn't this cause issues?