def forward_transfer(self, x):
    x = self.input_block(x)
    h = self.RNNCell(x, zeros_like(x))
    x = self.skip_1(x)
    h = self.RNNCell(x, h)
    x = self.skip_2(x)
    h = self.RNNCell(x, h)
    x = self.skip_3(x)
    h = self.RNNCell(x, h)
    return {'celltypes': self.celltype_out(h.detach()), 'modality 2': self.mod2_out(h), 'embedding': h}
def forward_transfer(self, x):
    x = self.input_block(x)
    h = self.RNNCell(x, zeros_like(x))
    x = self.skip_1(x)
    h = self.RNNCell(x, h)
    x = self.skip_2(x)
    h = self.RNNCell(x, h)
    x = self.skip_3(x)
    h = self.RNNCell(x, h)
    return {'celltypes': self.celltype_out(h.detach()), 'modality 2': self.mod2_out(h), 'embedding': h}