Каждая эпоха возвращает 0 с для train_loss, valid_loss и частоты ошибок - PullRequest
Новая
       121

Каждая эпоха возвращает 0 с для train_loss, valid_loss и частоты ошибок

0 голосов
/ 03 мая 2020

Я пытаюсь запустить 10 эпох на наборе данных Food-101 с помощью ResNet50, и пока это мой вывод: 

epoch   train_loss  valid_loss  error_rate  time
0   0.000000    0.000000    0.000000    21:05
1   0.000000    0.000000    0.000000    20:16
2   0.000000    0.000000    0.000000    20:17

Почему это может быть так?

Это мой код: 

from fastai import *
from fastai.vision import *
from pathlib import Path
from numba import vectorize
from subprocess import call, run
import os, git, glob, shutil

train_image_path = Path('images/train/')
test_image_path = Path('images/test/')
path = Path('../Food-101')

food_names = get_image_files(train_image_path)
file_parse = r'/([^/]+)_\d+\.(png|jpg|jpeg)$'

data = ImageDataBunch.from_folder(train_image_path, test_image_path, valid_pct=0.2, ds_tfms=get_transforms(), size=224)
data.normalize(imagenet_stats)

learn = cnn_learner(data, models.resnet50, metrics=error_rate)
learn.fit_one_cycle(10)
...