| 74 | self.error.emit(str(e)) |
| 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') |
| 89 | self.setGeometry(100, 100, 1200, 800) |
| 90 | |
| 91 | central_widget = QWidget() |
| 92 | self.setCentralWidget(central_widget) |
| 93 | |
| 94 | main_layout = QHBoxLayout() |
| 95 | central_widget.setLayout(main_layout) |
| 96 | |
| 97 | left_panel = self.create_left_panel() |
| 98 | right_panel = self.create_right_panel() |
| 99 | |
| 100 | main_layout.addWidget(left_panel, 1) |
| 101 | main_layout.addWidget(right_panel, 1) |
| 102 | |
| 103 | def create_left_panel(self): |
| 104 | panel = QWidget() |
| 105 | layout = QVBoxLayout() |
| 106 | panel.setLayout(layout) |
| 107 | |
| 108 | title = QLabel('🚀 NeuralForge Model Tester') |
| 109 | title.setFont(QFont('Arial', 20, QFont.Weight.Bold)) |
| 110 | title.setAlignment(Qt.AlignmentFlag.AlignCenter) |
| 111 | layout.addWidget(title) |
| 112 | |
| 113 | model_group = QGroupBox('Model Selection') |
| 114 | model_layout = QVBoxLayout() |
| 115 | |
| 116 | model_path_layout = QHBoxLayout() |
| 117 | self.model_path_input = QLineEdit() |
| 118 | self.model_path_input.setPlaceholderText('Path to model file (.pt)') |
| 119 | model_path_layout.addWidget(self.model_path_input) |
| 120 | |
| 121 | browse_btn = QPushButton('Browse') |
| 122 | browse_btn.clicked.connect(self.browse_model) |
| 123 | model_path_layout.addWidget(browse_btn) |
| 124 | |
| 125 | default_btn = QPushButton('Use Default') |
| 126 | default_btn.clicked.connect(self.use_default_model) |
| 127 | model_path_layout.addWidget(default_btn) |
| 128 | |
| 129 | model_layout.addLayout(model_path_layout) |
| 130 | |
| 131 | dataset_layout = QHBoxLayout() |
| 132 | dataset_label = QLabel('Dataset:') |
| 133 | self.dataset_input = QLineEdit('cifar10') |