bala1802 commited on
Commit
8c39932
1 Parent(s): 934a315

Update gpt_config.py

Browse files
Files changed (1) hide show
  1. gpt_config.py +2 -2
gpt_config.py CHANGED
@@ -6,8 +6,8 @@ block_size = 256 # what is the maximum context length for predictions?
6
  max_iters = 10000
7
  eval_interval = 500
8
  learning_rate = 3e-4
9
- # device = 'cuda' if torch.cuda.is_available() else 'cpu'
10
- device = 'mps'
11
  eval_iters = 200
12
  n_embd = 384
13
  n_head = 6
 
6
  max_iters = 10000
7
  eval_interval = 500
8
  learning_rate = 3e-4
9
+ device = 'cuda' if torch.cuda.is_available() else 'cpu'
10
+ # device = 'mps'
11
  eval_iters = 200
12
  n_embd = 384
13
  n_head = 6