Spaces:
Runtime error
Runtime error
Update S10.py
Browse files
S10.py
CHANGED
|
@@ -226,5 +226,9 @@ class Cifar10SearchDataset(pl.LightningModule):
|
|
| 226 |
return DataLoader(self.cifar10_test, batch_size=64)
|
| 227 |
|
| 228 |
|
| 229 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 230 |
|
|
|
|
| 226 |
return DataLoader(self.cifar10_test, batch_size=64)
|
| 227 |
|
| 228 |
|
| 229 |
+
model = Cifar10SearchDataset()
|
| 230 |
+
trainer = pl.Trainer(accelerator="auto",max_epochs=24)
|
| 231 |
+
#datasetcifar=Cifar10SearchDataset()
|
| 232 |
+
#trainer = pl.Trainer()
|
| 233 |
+
trainer.fit(model)
|
| 234 |
|