ReLSO-Guided-Generative-Protein-Design-using-Regularized-Transformers
ReLSO-Guided-Generative-Protein-Design-using-Regularized-Transformers copied to clipboard
possibly incompatible dimensionality in pooling.py
In pooling.py there is self.fc1 which has output dim = input_dim//2 and self.glob_attn_module with input dim = input_dim. In forward step there is an if condition that applies self.fc1 and then connects its output to the self.glob_attn_module which can lead to an error due to different dimensionality.