update regressor structure

regressor.py

@@ -6,10 +6,13 @@ class WRegressor(nn.Module):
         self.linear_relu_stack = nn.Sequential(
             nn.Linear(768, 256),
             nn.ReLU(),
+            nn.Dropout(),
             nn.Linear(256, 64),
             nn.ReLU(),
+            nn.Dropout(),
             nn.Linear(64, 16),
             nn.ReLU(),
+            nn.Dropout(),
             nn.Linear(16, 1),
         )
         return