Hugging Face's logo

Spaces:
cycool29
/
SpiralSense
Runtime error

App Files Community
cycool29 commited on
Commit
59d6532
·
1 Parent(s): e2b5593

Update

Browse files
Files changed (2) hide show
  1. configs.py +3 -3
  2. requirements.txt +0 -0
configs.py CHANGED
@@ -8,11 +8,11 @@ from torchvision.models import squeezenet1_0, SqueezeNet1_0_Weights
8
  from torchvision.models import squeezenet1_0
9
  # Constants
10
  RANDOM_SEED = 123
11
- BATCH_SIZE = 32
12
  NUM_EPOCHS = 40
13
- LEARNING_RATE = 0.00017588413773574044
14
  STEP_SIZE = 10
15
- GAMMA = 0.9
16
  DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
17
  NUM_PRINT = 100
18
  TASK = 1
 
8
  from torchvision.models import squeezenet1_0
9
  # Constants
10
  RANDOM_SEED = 123
11
+ BATCH_SIZE = 16
12
  NUM_EPOCHS = 40
13
+ LEARNING_RATE = 5.488903014780378e-05
14
  STEP_SIZE = 10
15
+ GAMMA = 0.3
16
  DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
17
  NUM_PRINT = 100
18
  TASK = 1
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ