I was trying to test the examples you put in the repository (torch_examples.ipynb), but several things did not work for me, so I get the impression that the project has done some refactoring and this notebook has not been updated.
The first thing is that the CustomModel type is not implemented in model_torch.py. Then the second problem I encountered is that the ConvBlockModel type imports ConvBlock type is not implemented in batchflow.models.torch.layers.
What causes me more doubt about these errors is that the ConvBlock type has not been implemented since a few releases ago, they deleted it and do not plan to use it anymore? In that case, they would have to resolve the layouts in the ConvBlockModel type definition.
I was trying to test the examples you put in the repository (torch_examples.ipynb), but several things did not work for me, so I get the impression that the project has done some refactoring and this notebook has not been updated.
The first thing is that the
CustomModeltype is not implemented inmodel_torch.py. Then the second problem I encountered is that theConvBlockModeltype importsConvBlocktype is not implemented inbatchflow.models.torch.layers.What causes me more doubt about these errors is that the
ConvBlocktype has not been implemented since a few releases ago, they deleted it and do not plan to use it anymore? In that case, they would have to resolve the layouts in theConvBlockModeltype definition.