From f1368971ff201ad9cd7bcc7f320eafc84a0aa464 Mon Sep 17 00:00:00 2001 From: Florin Ratajczak Date: Mon, 14 Oct 2024 11:21:18 +0000 Subject: [PATCH] removed layers --- speos/architectures.py | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/speos/architectures.py b/speos/architectures.py index 44d24ea..e037e41 100644 --- a/speos/architectures.py +++ b/speos/architectures.py @@ -8,7 +8,6 @@ from speos.utils.logger import setup_logger from speos.utils.config import Config import speos.utils.nn_utils as nn_utils -import speos.layers as layers class LINKX(nn.Module): @@ -244,20 +243,8 @@ def get_mp_layer(self, i): if self.gcnconv_parameters["type"] == "rgcn": mp_layer = pyg_nn.RGCNConv(self.gcnconv_parameters["dim"], self.gcnconv_parameters["dim"], self.num_adjacencies, **kwargs) - if self.gcnconv_parameters["type"] == "rtag": - mp_layer = layers.RTAGConv(self.gcnconv_parameters["dim"], self.gcnconv_parameters["dim"], self.num_adjacencies, **kwargs) - elif self.gcnconv_parameters["type"] == "rgat": - mp_layer = layers.RGATConv(self.gcnconv_parameters["dim"], self.gcnconv_parameters["dim"], self.num_adjacencies, **kwargs) - elif self.gcnconv_parameters["type"] == "rgattag": - mp_layer = layers.RGATTAGConv(self.gcnconv_parameters["dim"], self.gcnconv_parameters["dim"], self.num_adjacencies, **kwargs) elif self.gcnconv_parameters["type"] == "film": mp_layer = pyg_nn.FiLMConv(self.gcnconv_parameters["dim"], self.gcnconv_parameters["dim"], self.num_adjacencies, **kwargs) - elif self.gcnconv_parameters["type"] == "filmtag": - mp_layer = layers.FiLMTAGConv(self.gcnconv_parameters["dim"], self.gcnconv_parameters["dim"], self.num_adjacencies, **kwargs) - elif self.gcnconv_parameters["type"] == "mlpfilm": - mp_layer = layers.MLPFiLM(self.gcnconv_parameters["dim"], self.gcnconv_parameters["dim"], self.num_adjacencies, **kwargs) - elif self.gcnconv_parameters["type"] == "filmfilm": - mp_layer = layers.FiLMFiLM(self.gcnconv_parameters["dim"], self.gcnconv_parameters["dim"], self.num_adjacencies, **kwargs) else: try: