Import Error

[omariosc]

Missing line from 02_monai_components.ipynb.

from monai.data import CacheDataset

Full cell:

from monai.networks.nets import UNet
from monai.losses import DiceLoss
from monai.engines import SupervisedTrainer
from monai.data import CacheDataset # Missing line

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
network = UNet(3,1,3,[8,16,32],[2,2,2]).to(device)
loss = DiceLoss(softmax=True,to_onehot_y=True)

ds = CacheDataset(filenames, transform=trans) # Does not run
dl = DataLoader(ds,batch_size=6, num_workers=8)

trainer=SupervisedTrainer(
    device=device,
    max_epochs=4,
    train_data_loader=dl,
    network=network,
    optimizer=torch.optim.Adam(network.parameters()),
    loss_function=loss
)