frameworks

2024-03-06 22:40:09 +08:00
parent a0500f5111
commit ea146f8302
5 changed files with 296 additions and 0 deletions
@@ -10,6 +10,11 @@ M = 16

 class Net(nn.Module):
    def __init__(self, n_input_channels, n_actions):
+        ## input channels 是数据的输入的维度
+        ## 每个状态是一个
+        ## N * N * (M_c + 1)的tensor
+        ## 其中 N * N 代表的是用户的数量 如果有1000个用户 大概为32
+        ## 第M_c个matrix代表的是当前用户的申请或是离开
        super(Net, self).__init__()
        self.conv1 = nn.Conv2d(in_channels=n_input_channels, out_channels=16, kernel_size=5, stride=1, padding=0)
        self.conv2 = nn.Conv2d(in_channels=16, out_channels=32, kernel_size=3, stride=1, padding=0)