Я недавно начал работать с Google Colab для участия в соревнованиях Kaggle Airbus. Ранее я писал свой код на своем компьютере и запускал его локально, но вычислительных ресурсов стало не хватать - я перешел на Google Colab
Я начал с клонирования репо, используя !git clone http://url/repo.git, и все прошло нормально, мои данные были вставлены в папку /content/repo. Я следовал инструкциям по копированию данных из kaggle в Google Colab напрямую, без загрузки и загрузки.
Структура файлов:
/
- Content:
--- Airbus_project:
------ train.py
------ input:
----------- train_data (folder with images)
------ keras_frcnn:
----------- data_generators.py
----------- config.py
Я также увидел, что запуск !python3 train.py не работает, если я хочу внести изменения в мой код (и !git pull не работает), поэтому я скопировал код из файла train.py в одну ячейку ноутбука.
Ячейка, конечно, начинается с импорта некоторых модулей
from __future__ import division
import random
(...)
import os
import tensorflow as tf
from keras import backend as K
from keras.optimizers import Adam
from keras.layers import Input
from keras.models import Model
from keras_frcnn import config, data_generators
(...)
И когда я пытаюсь запустить его, я получаю ошибку ModuleNotFoundError: No module named 'keras_frcnn'. Я искал помощи по этому вопросу, и большинство результатов предлагают использовать Google Drive и добавить
from google.colab import drive
drive.mount('/content/gdrive')
Но я не могу сохранить свои данные на Google Диске из-за нехватки места (данные Kaggle составляют около 35 ГБ).
Можно ли как-то настроить Google Colab для поиска этих модулей в папке keras_frcnn, которая находится в той же папке, что и файл train.py?