ReLSO-Guided-Generative-Protein-Design-using-Regularized-Transformers icon indicating copy to clipboard operation
ReLSO-Guided-Generative-Protein-Design-using-Regularized-Transformers copied to clipboard

possibly incompatible dimensionality in pooling.py

Open pailozka opened this issue 1 year ago • 0 comments

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.

pailozka avatar Aug 13 '24 19:08 pailozka