Spaces:

Transduce
/

testing_cuda

Build error

Transduce commited on Jun 28, 2024

Commit

864e17f

verified ·

1 Parent(s): 6fffa05

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ class SimpleNet(nn.Module):
         x = self.fc(x)
         return x
-@spaces.GPU(duration=120)
 def train_model(epochs):
     # Load MNIST dataset
     transform = transforms.Compose([transforms.ToTensor()])

         x = self.fc(x)
         return x
+@spaces.GPU(duration=121)
 def train_model(epochs):
     # Load MNIST dataset
     transform = transforms.Compose([transforms.ToTensor()])