davesalvi commited on
Commit
13257bc
·
1 Parent(s): 4e083fb

clean code

Browse files
Files changed (2) hide show
  1. .idea/workspace.xml +1 -4
  2. src/lcnn_model.py +1 -5
.idea/workspace.xml CHANGED
@@ -5,9 +5,6 @@
5
  </component>
6
  <component name="ChangeListManager">
7
  <list default="true" id="23565123-73ab-4f40-a9ef-1086e0c9e1ec" name="Changes" comment="">
8
- <change afterPath="$PROJECT_DIR$/src/moe_model.py" afterDir="false" />
9
- <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
10
- <change beforePath="$PROJECT_DIR$/script.py" beforeDir="false" afterPath="$PROJECT_DIR$/script.py" afterDir="false" />
11
  <change beforePath="$PROJECT_DIR$/src/lcnn_model.py" beforeDir="false" afterPath="$PROJECT_DIR$/src/lcnn_model.py" afterDir="false" />
12
  </list>
13
  <option name="SHOW_DIALOG" value="false" />
@@ -134,7 +131,7 @@
134
  <workItem from="1742889011586" duration="41000" />
135
  <workItem from="1742890414523" duration="4382000" />
136
  <workItem from="1742941362366" duration="1283000" />
137
- <workItem from="1742970904084" duration="1559000" />
138
  </task>
139
  <servers />
140
  </component>
 
5
  </component>
6
  <component name="ChangeListManager">
7
  <list default="true" id="23565123-73ab-4f40-a9ef-1086e0c9e1ec" name="Changes" comment="">
 
 
 
8
  <change beforePath="$PROJECT_DIR$/src/lcnn_model.py" beforeDir="false" afterPath="$PROJECT_DIR$/src/lcnn_model.py" afterDir="false" />
9
  </list>
10
  <option name="SHOW_DIALOG" value="false" />
 
131
  <workItem from="1742889011586" duration="41000" />
132
  <workItem from="1742890414523" duration="4382000" />
133
  <workItem from="1742941362366" duration="1283000" />
134
+ <workItem from="1742970904084" duration="1716000" />
135
  </task>
136
  <servers />
137
  </component>
src/lcnn_model.py CHANGED
@@ -24,12 +24,8 @@ class LCNN(nn.Module):
24
  def __init__(self, return_emb=False, num_class=2):
25
  super(LCNN, self).__init__()
26
 
27
- <<<<<<< HEAD
28
- self.threshold = 0.7
29
- self.return_emb = return_emb
30
- =======
31
  self.threshold = 0.5
32
- >>>>>>> 8ce022751ede24f85b3e26af79e0745f8462b868
33
 
34
  # Feature Extraction Part (First Part)
35
  self.dropout1 = nn.Dropout(0.2)
 
24
  def __init__(self, return_emb=False, num_class=2):
25
  super(LCNN, self).__init__()
26
 
 
 
 
 
27
  self.threshold = 0.5
28
+ self.return_emb = return_emb
29
 
30
  # Feature Extraction Part (First Part)
31
  self.dropout1 = nn.Dropout(0.2)