62cea7fcfcc9e20b784e08bddce5dc69b9578ef4,tensorforce/core/baselines/cnn_baseline.py,CNNBaseline,__init__,#CNNBaseline#Any#Any#Any#Any#Any#,24

Before Change

            conv_sizes: List of convolutional layer sizes
            dense_sizes: List of dense layer sizes
        
        network_spec = list()

        // Convolution layers
        for size in conv_sizes:
            network_spec.append(dict(type="conv2d", size=size))

        // First layer with larger window
        network_spec[0]["window"] = 5

        // Global max-pooling
        network_spec.append(dict(type="global_pooling", pooling="max"))

        // Dense layers
        for size in dense_sizes:
            network_spec.append(dict(type="dense", size=size))

After Change

        for size in conv_sizes:
            network.append(dict(type="conv2d", size=size))

        // First layer with larger window
        network[0]["window"] = 5

        // Global max-pooling
        network.append(dict(type="pooling", pooling="max"))

        // Dense layers
        for size in dense_sizes:
            network.append(dict(type="dense", size=size))

        super().__init__(
            name=name, network=network, inputs_spec=inputs_spec,
            l2_regularization=l2_regularization, summary_labels=summary_labels
        )

Instances