Page MenuHomePhabricator | MITK

Problem with numpy dataset and json serialization in PyTorchExperiment class
Open, NormalPublic

Description

run() method in Experiment.py aborts because
save_config() method in ExperimentLogger.py fails for

Numpy array containing the image dataset, wrapped in pytorch TensorDataset, DataLoader

Output: TypeError: Object of type 'DataLoader' is not JSON serializable

Event Timeline

kausch created this task.Apr 9 2018, 4:35 PM
petersej triaged this task as Normal priority.Thu, Jan 24, 11:30 AM
petersej added a subscriber: petersej.

Does PyTorch offer a way to serialize DataLoader objects? If so, we could extend ModuleMultiTypeEncoder/Decoder in util.util