ankxyz

Error on save PyTorch model defined in another module

12

I defined custom PyTorch network. If I try to save it using MLEM, it fails. **Environment** *OS*: Ubuntu Linux 20.04 *Python*: 3.8.10 *Virtual envronment*: venv *Python dependencies*: - numpy==1.22.4 -...

I want to add to each sample of my dataset some label with several field: ```python for sample in dataset: sample['my_label'] = { 'field_1': [