Minor update for DQN

This commit is contained in:
Shangtong Zhang committed 2017-12-21 22:22:36 -07:00
1 parent 8839614c44
commit 69b7ea53cc
5 files changed
+17 -31

No files matched your search

-2
View File
@@ -13,8 +13,6 @@ import numpy as np
# Base class for all kinds of network
class BasicNet:
def __init__(self, optimizer_fn, gpu, LSTM=False):
if optimizer_fn is not None:
self.optimizer = optimizer_fn(self.parameters())
self.gpu = gpu and torch.cuda.is_available()
self.LSTM = LSTM
if self.gpu: