Skip to content

Commit

Permalink
Changing DRPVAL to 0.1 to match milestone experiments config
Browse files Browse the repository at this point in the history
  • Loading branch information
Jatin Sharma committed Mar 18, 2021
1 parent 560ced2 commit 9fd42a1
Showing 1 changed file with 10 additions and 8 deletions.
18 changes: 10 additions & 8 deletions ITrackerModel.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@
}
'''

# Global Dropout value
DRPVAL = 0.1

class ItrackerImageModel(nn.Module):
# Used for both eyes (with shared weights) and the face (with unique weights)
Expand Down Expand Up @@ -70,13 +72,13 @@ def __init__(self, color_space, model_type):
self.fc = nn.Sequential(
# FC-F1
# 25088
nn.Dropout(0.4),
nn.Dropout(DRPVAL),
nn.Linear(25088, 128),
# 128
nn.ReLU(inplace=True),

# FC-F2
nn.Dropout(0.4),
nn.Dropout(DRPVAL),
nn.Linear(128, 64),
# 64
nn.ReLU(inplace=True),
Expand All @@ -98,13 +100,13 @@ def __init__(self, color_space, model_type):
self.fc = nn.Sequential(
# FC-F1
# 25088
nn.Dropout(0.4),
nn.Dropout(DRPVAL),
nn.Linear(25088, 256),
# 256
nn.ReLU(inplace=True),

# FC-F2
nn.Dropout(0.4),
nn.Dropout(DRPVAL),
nn.Linear(256, 128),
# 128
nn.ReLU(inplace=True),
Expand Down Expand Up @@ -133,7 +135,7 @@ def __init__(self, gridSize=25):

# FC-FG2
# 256
nn.Dropout(0.4),
nn.Dropout(DRPVAL),
nn.Linear(256, 128),
# 128
nn.ReLU(inplace=True),
Expand Down Expand Up @@ -162,7 +164,7 @@ def __init__(self, color_space, model_type):
# Joining both eyes
self.eyesFC = nn.Sequential(
# FC-E1
nn.Dropout(0.4),
nn.Dropout(DRPVAL),
# 50176
nn.Linear(2 * 25088, 128),
# 128
Expand All @@ -173,7 +175,7 @@ def __init__(self, color_space, model_type):
# Joining everything
self.fc = nn.Sequential(
# FC1
nn.Dropout(0.4),
nn.Dropout(DRPVAL),
# 384 FC-E1 (128) + FC-F2(64) + FC-FG2(128)
nn.Linear(128 + 64 + 128, 128),
# 128
Expand All @@ -182,7 +184,7 @@ def __init__(self, color_space, model_type):

# FC2
# 128
nn.Dropout(0.4),
nn.Dropout(DRPVAL),
nn.Linear(128, 2),
# 2
)
Expand Down

0 comments on commit 9fd42a1

Please sign in to comment.