clean code
Browse files- .idea/workspace.xml +1 -4
- 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="
|
| 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 |
-
|
| 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)
|