looked like a typo (#18)

* looked like a typo * adding .idea to gitignore * removing file * deleting folder * idea workspace removal
kexinhuang12345 · Aug 18, 2020 · 65e562c · 65e562c
1 parent 97a9edd
commit 65e562c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,5 +1,6 @@
 .ipynb_checkpoints
 .DS_Store
+.idea
 
 __pycache__/
 

diff --git a/DeepPurpose/models.py b/DeepPurpose/models.py
@@ -137,7 +137,7 @@ def __init__(self, encoding, **config):
 				raise AttributeError('Please use LSTM or GRU.')
 			direction = 2 if config['rnn_drug_bidirectional'] else 1
 			self.rnn = self.rnn.double()
-			self.fc1 = nn.Linear(config['rnn_drug_hid_dim'] * direction * n_size_p, config['hidden_dim_drug'])
+			self.fc1 = nn.Linear(config['rnn_drug_hid_dim'] * direction * n_size_d, config['hidden_dim_drug'])
 
 		if encoding == 'protein':
 			in_ch = [26] + config['cnn_target_filters']