def __init__(self, opt, kernel_size=2, n_blocks=1, state_dim_bottleneck=64, annotation_dim_bottleneck=64): super(STGCN, self).__init__() self.batchSize = opt.batchSize self.state_dim = opt.state_dim self.n_node = opt.n_node # ST-Block self.st_blocks = ST_blocks(opt, kernel_size=kernel_size, n_blocks=n_blocks, state_dim_bottleneck=state_dim_bottleneck, annotation_dim_bottleneck=annotation_dim_bottleneck) opt.L = opt.L - 2 * n_blocks * (kernel_size - 1) # GCNN () self.gcnn = GatedCNN(opt, in_channels=self.state_dim, out_channels=self.state_dim, kernel_size=opt.L) opt.L = 1 # FC () self.out = nn.Sequential( nn.Linear(self.state_dim, opt.output_dim), nn.Sigmoid() ) opt.L = opt.init_L self._initialization()