Skip to content

Commit cd6ca78

Browse files
committed
couple tweaks
1 parent 6f2608c commit cd6ca78

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

agent.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,12 @@
55
from loading_anim import LoadingAnim
66
from random import shuffle
77

8-
Data = ['30pgames.pt', 'data_21k_42.pt']
8+
Data = ['50pgames.pt', 'data_21k_42.pt']
99
Layers = [64, 512, 512, 512, 128, 3]
1010
Epochs = 500
1111
BatchSize = 1000
1212
LearnRate = 0.005
13-
ModelFile = 'model_ns.pt'
13+
ModelFile = 'model.pt'
1414

1515
device = t.device("cuda" if t.cuda.is_available() else "cpu")
1616
print(f"Using {device} ")

snake_user.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
game_size = 8 # has to be same size as version NN plays
1313
folder = 'mygames'
14-
ham = 0 # no hamiltonian, goes straight on blank input
14+
ham = False # no hamiltonian, goes straight on blank input
1515
#ham = [1,1,1,2,0,1,1,1] #for Hamiltonian
1616
#ham = [1,1,1,1,2,2,1,1,1,1,1,1,0,0,1,1] #for zigzag
1717
#ham = [1,1,1,1,2,1,1,2,1,2,2,0,0,1,0,1,1,1,1,1,0,1,1,0,1,0,0,2,2,1,2,1] #4quad

0 commit comments

Comments
 (0)