(self)
| 75 | |
| 76 | class NeuralForgeGUI(QMainWindow): |
| 77 | def __init__(self): |
| 78 | super().__init__() |
| 79 | self.model = None |
| 80 | self.device = 'cuda' if torch.cuda.is_available() else 'cpu' |
| 81 | self.classes = [] |
| 82 | self.dataset_name = 'cifar10' |
| 83 | |
| 84 | self.init_ui() |
| 85 | self.apply_stylesheet() |
| 86 | |
| 87 | def init_ui(self): |
| 88 | self.setWindowTitle('NeuralForge - Model Tester') |
no test coverage detected