SahithiR commited on
Commit
20d8eca
·
1 Parent(s): 3e8b548

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -18,11 +18,11 @@ from torchmetrics import Accuracy
18
  from torch.nn import functional as F
19
  import matplotlib.pyplot as plt
20
 
21
- model = Cifar10SearchDataset()
22
- trainer = pl.Trainer(accelerator="auto",max_epochs=24)
23
 
24
- trainer.fit(model)
25
- trainer.test(model)
26
 
27
 
28
  #model = MyLightningModule()
 
18
  from torch.nn import functional as F
19
  import matplotlib.pyplot as plt
20
 
21
+ #model = Cifar10SearchDataset()
22
+ #trainer = pl.Trainer(accelerator="auto",max_epochs=24)
23
 
24
+ #trainer.fit(model)
25
+ #trainer.test(model)
26
 
27
 
28
  #model = MyLightningModule()