Skip to content
Snippets Groups Projects
Commit 9b430d78 authored by Yandi's avatar Yandi
Browse files

deeper cnn

parent c9b529d1
No related branches found
No related tags found
1 merge request!1Master into main
No preview for this file type
......@@ -121,12 +121,13 @@ class CNN1D(torch.nn.Module):
super(CNN1D, self).__init__()
self.model = torch.nn.Sequential(
*conv_block(num_inputs, 5, 0.01),
*conv_block(32, 6, 0.01)
*conv_block(32, 6, 0.1),
*conv_block(64,7,0.01)
)
self.avg_pool = torch.nn.AdaptiveAvgPool1d(1)
self.fc = nn.Sequential(
nn.Linear(64, 32),
nn.Linear(128, 32),
nn.ReLU(),
nn.Linear(32,cfg["Dataset"]["num_days"])
)
......
No preview for this file type
No preview for this file type
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment