Method
__init__
(
self,
experiment_id,
model,
train_loader,
val_loader,
test_loader,
epochs,
learning_rate,
patience,
general_hyperparameters,
model_hyperparameters,
)
Source from the content-addressed store, hash-verified
| 233 | |
| 234 | class BatchGDManager: |
| 235 | def __init__( |
| 236 | self, |
| 237 | experiment_id, |
| 238 | model, |
| 239 | train_loader, |
| 240 | val_loader, |
| 241 | test_loader, |
| 242 | epochs, |
| 243 | learning_rate, |
| 244 | patience, |
| 245 | general_hyperparameters, |
| 246 | model_hyperparameters, |
| 247 | ): |
| 248 | self.experiment_id = experiment_id |
| 249 | self.model = model |
| 250 | self.train_loader = train_loader |
| 251 | self.val_loader = val_loader |
| 252 | self.test_loader = test_loader |
| 253 | self.epochs = epochs |
| 254 | self.learning_rate = learning_rate |
| 255 | self.patience = patience |
| 256 | self.general_hyperparameters = general_hyperparameters |
| 257 | self.model_hyperparameters = model_hyperparameters |
| 258 | self.lob_lightning_module = None |
| 259 | self.trainer = None |
| 260 | self.deleted_run = None |
| 261 | |
| 262 | def delete_run(self): |
| 263 | api = wandb.Api() |
Tested by
no test coverage detected