nateraw

/

test-classifier-flash-2

like 0

• "torchvision.transforms.transforms.ToTensor",
• "torch._C._VariableFunctionsClass",
• "flash.core.data.process.Postprocess",
• "torch._utils._rebuild_tensor_v2",
• "torch.optim.adam.Adam",
• "torch.nn.modules.container.Sequential",
• "torch.distributions.bernoulli.Bernoulli",
• "torch.nn.modules.container.ModuleDict",
• "kornia.augmentation.augmentation.RandomHorizontalFlip",
• "flash.core.finetuning.Freeze",
• "torch.utils.data._utils.collate.default_collate",
• "torch.device",
• "torch.BoolStorage",
• "flash.core.data.data_pipeline.DataPipeline",
• "flash.image.data.image_loader",
• "torch.Size",
• "flash.core.data.data_source.DefaultDataKeys",
• "flash.image.data.ImageTensorDataSource",
• "flash.image.classification.data.ImageClassificationDataFrameDataSource",
• "kornia.augmentation.augmentation.Normalize",
• "flash.image.classification.data.MatplotlibVisualization",
• "collections.OrderedDict",
• "flash.core.classification.Labels",
• "flash.image.data.ImageNumpyDataSource",
• "flash.image.data.ImagePathsDataSource",
• "torch.float32",
• "flash.core.data.transforms.kornia_collate",
• "flash.image.classification.data.ImageClassificationPreprocess",
• "torch.FloatStorage",
• "torch.LongStorage",
• "flash.image.data.ImageDeserializer",
• "kornia.geometry.transform.affwarp.Resize",
• "__builtin__.set",
• "flash.core.data.transforms.ApplyToKeys",
• "flash.core.data.data_source.LabelsState",
• "__builtin__.getattr",
• "flash.image.data.ImageFiftyOneDataSource",
• "flash.core.data.data_source.DatasetDataSource",
• "pytorch_lightning.callbacks.model_checkpoint.ModelCheckpoint",
• "flash.core.data.data_pipeline.DataPipelineState",
• "flash.core.data.utils.FuncModule",
• "flash.core.data.data_source.DefaultDataSources"

How to fix it?