Witam,
uczę się deep learningu. Próbuję swoich sił w konkursie kaggle. Niestety mój GPU to tylko GTX 1050. Mój problem jest taki, że maksymalny batch size to 4. Który nie powoduje błędu. Czy da się jakoś to poprawić? Korzystam z MASK_RCNN.
# Train on 1 GPU and 8 images per GPU. We can put multiple images on each
# GPU because the images are small. Batch size is 8 (GPUs * images/GPU).
GPU_COUNT = 1
IMAGES_PER_GPU = 8
BACKBONE = 'resnet50'
NUM_CLASSES = 2 # background + 1 pneumonia classes
IMAGE_MIN_DIM = 256
IMAGE_MAX_DIM = 256
RPN_ANCHOR_SCALES = (16, 32, 64, 128)
TRAIN_ROIS_PER_IMAGE = 32
Jeżeli zmienie 256 na 128 też wyskakują mi jakieś błedy i uczenie się przerywa.
Wiem, że kernele w kaggle dają możliwość korzystania z ich GPU. Ale niestety 6h trwa... a w polowie wyskoczy jakiś błąd, żeby odswiezyc skrypt. To czlowieka krew zalewa...
Z góry dziękuje :)