cnngeometric_pytorch
cnngeometric_pytorch copied to clipboard
Fixed a bug (error) in the Demo Notebook
The CNNGeometric Class does not have a parameter for the Geometric Model, and the model does not run successfully, and an error is thrown upon running the cells.
Instead, I have passed the output size for the desired geometric model which makes our code (notebook) work successfully.