Removed stray device
@@ -23,7 +23,6 @@ class Nucleus ():
self._config = config
self._forward_pool = ThreadPoolExecutor(maxsize = self._config.nucleus.queue_maxsize, max_workers=self._config.nucleus.max_workers)
self._backward_pool = ThreadPoolExecutor(maxsize = self._config.nucleus.queue_maxsize, max_workers=self._config.nucleus.max_workers)
- self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
def forward(self, synapse: Synapse, inputs: torch.Tensor, mode: bittensor_pb2.Modality, priority: float) -> Tuple[torch.FloatTensor, str, int]:
r""" Accepts a synapse object with inputs and priority, submits them to the forward work pool
