Fix the things that I broke

Author: mahowald

Dockerfile

@@ -6,7 +6,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY setup.py .
 COPY tictactoe/ ./tictactoe/
 RUN python setup.py install
-COPY pytorch_dqn-v6.pt .
+COPY pytorch_dqn.pt .
 
 ENV TERM=xterm-color
 

tictactoe/play.py

@@ -14,7 +14,7 @@
 def play():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     env = TicTacToe()
-    model = load_model("pytorch_dqn-v6.pt", device)
+    model = load_model("pytorch_dqn.pt", device)
 
     done = False
     obs = env.reset()