atom view #42

I was wondering how atom view is implemented? I'm getting a shape mismatch.

In mc-gearnet_edge.yaml I changed the view and entity level to 'atom' and input dimension to 38. As i find 38 atom types in the torchdrug protein class.
Is there another setting i need to change?

Thanks for creating GearNet!