Fixed imports
This commit is contained in:
@@ -35,7 +35,7 @@ class FeedForwardNetwork(nn.Module):
|
||||
x = self.__relu(x)
|
||||
|
||||
# 3) Dropout
|
||||
x = self.__dropout(x)
|
||||
# x = self.__dropout(x)
|
||||
|
||||
# 4) Linear Layer
|
||||
x = self.__fully_connected_2(x)
|
||||
|
||||
Reference in New Issue
Block a user