Fastai v2 vision

[muellerzr (Zachary Mueller)]

So again this probably has to do with the y’s (this is when I saw these errors popping up). We discussed this in the study group (but it’s buried). Try incorporating the ideas from this thread to make sure you didn’t miss any classes:

CUDA error: device-side assert triggered

What happens if you execute: import torch as pt mask_values = pt.unique(mask.data) and compare the length of mask_values to codes ? My assumption is that there is a mismatch between the pixels values and the codes. If that doesn’t help, can you please replace 400 in the following line with src_size: data = (il.transform(get_transforms(), tfm_y=True,size=400) .databunch(bs=2) .normalize(imagenet_stats))