Загрузка предварительно обученных моделей FastAI в ядра Kaggle без использования Интернета

Question

Я пытаюсь загрузить модель densenet121 в ядро ​​Kaggle, не включая Интернет. Я выполнил необходимые шаги, такие как добавление предварительно обученных весов в свой входной каталог и перемещение его в '.cache/torch/checkpoints/'. Все равно не работает и выкидывает гайеррор.

Ниже приводится код SNIPPET:

!mkdir -p /tmp/.cache/torch/checkpoints
!cp ../input/fastai-pretrained-models/densenet121-a639ec97.pth /tmp/.cache/torch/checkpoints/densenet121-a639ec97.pth

learn_cd = create_cnn(data_cd, models.densenet121, metrics=[error_rate, accuracy],model_dir = Path('../kaggle/working/models'),path=Path('.'),).to_fp16()

Я долго с этим борюсь. Любая помощь будет чрезвычайно полезна

Answer 1

Поэтому входной путь "../input/" в ядре kaggle доступен только для чтения. скорее создайте папку в "kaggle/working" и скопируйте туда веса модели. Пример ниже

if not os.path.exists('/root/.cache/torch/hub/checkpoints/'):
        os.makedirs('/root/.cache/torch/hub/checkpoints/')

!mkdir '/kaggle/working/resnet34'
!cp '/root/.cache/torch/hub/checkpoints/resnet34-333f7ec4.pth' '/kaggle/working/resnet34/resnet34.pth'