models/s2s_big_baseline.py [124:167]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        ])

        self.det_init_net = nn.Sequential(
            layers.DcConv(2*enc_dim*self.n_ctx, 2*enc_dim*self.n_ctx, 1),
            layers.TemporalConv2d(2*enc_dim*self.n_ctx, 2*enc_dim, 1),
            layers.TemporalNorm2d(1, 2*enc_dim),
        )

        self.prior_init_nets = nn.ModuleDict({
            'layer_4': nn.Sequential(
                layers.DcConv(enc_dim*self.n_ctx, enc_dim*self.n_ctx, 1),
                layers.TemporalConv2d(enc_dim*self.n_ctx, enc_dim*2, 1),
                layers.TemporalNorm2d(1, 2*enc_dim),
            ),
        })

        self.posterior_init_nets = nn.ModuleDict({
            'layer_4': nn.Sequential(
                layers.DcConv(enc_dim*self.n_ctx, enc_dim*self.n_ctx, 1),
                layers.TemporalConv2d(enc_dim*self.n_ctx, enc_dim*2, 1),
                layers.TemporalNorm2d(1, 2*enc_dim),
            ),
        })

        self.posterior_branches = nn.ModuleDict({
            'layer_4': nn.ModuleList([
                layers.TemporalConv2d(enc_dim, n_z, 1),
                layers.TemporalNorm2d(1, n_z),
                layers.ConvLSTM(n_z, enc_dim, norm=True),
                layers.TemporalConv2d(enc_dim, n_z*2, 1),
            ])
        })

        self.prior_branches = nn.ModuleDict({
            'layer_4': nn.ModuleList([
                layers.TemporalConv2d(enc_dim, n_z, 1),
                layers.TemporalNorm2d(1, n_z),
                layers.ConvLSTM(n_z, enc_dim, norm=True),
                layers.TemporalConv2d(enc_dim, n_z*2, 1),
            ])
        })

        # Connection list
        self.det_init_connections = {
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



models/s2s_convlstm_baseline.py [65:108]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        ])

        self.det_init_net = nn.Sequential(
            layers.DcConv(2*enc_dim*self.n_ctx, 2*enc_dim*self.n_ctx, 1),
            layers.TemporalConv2d(2*enc_dim*self.n_ctx, 2*enc_dim, 1),
            layers.TemporalNorm2d(1, 2*enc_dim),
        )

        self.prior_init_nets = nn.ModuleDict({
            'layer_4': nn.Sequential(
                layers.DcConv(enc_dim*self.n_ctx, enc_dim*self.n_ctx, 1),
                layers.TemporalConv2d(enc_dim*self.n_ctx, enc_dim*2, 1),
                layers.TemporalNorm2d(1, 2*enc_dim),
            ),
        })

        self.posterior_init_nets = nn.ModuleDict({
            'layer_4': nn.Sequential(
                layers.DcConv(enc_dim*self.n_ctx, enc_dim*self.n_ctx, 1),
                layers.TemporalConv2d(enc_dim*self.n_ctx, enc_dim*2, 1),
                layers.TemporalNorm2d(1, 2*enc_dim),
            ),
        })

        self.posterior_branches = nn.ModuleDict({
            'layer_4': nn.ModuleList([
                layers.TemporalConv2d(enc_dim, n_z, 1),
                layers.TemporalNorm2d(1, n_z),
                layers.ConvLSTM(n_z, enc_dim, norm=True),
                layers.TemporalConv2d(enc_dim, n_z*2, 1),
            ])
        })

        self.prior_branches = nn.ModuleDict({
            'layer_4': nn.ModuleList([
                layers.TemporalConv2d(enc_dim, n_z, 1),
                layers.TemporalNorm2d(1, n_z),
                layers.ConvLSTM(n_z, enc_dim, norm=True),
                layers.TemporalConv2d(enc_dim, n_z*2, 1),
            ])
        })

        # Connection list
        self.det_init_connections = {
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -