Fashion-MNIST 2c2d¶
-
class
deepobs.pytorch.testproblems.fmnist_2c2d.
fmnist_2c2d
(batch_size, weight_decay=None)[source]¶ DeepOBS test problem class for a two convolutional and two dense layered neural network on Fashion-MNIST.
The network has been adapted from the TensorFlow tutorial and consists of
- two conv layers with ReLUs, each followed by max-pooling
- one fully-connected layers with ReLUs
- 10-unit output layer with softmax
- cross-entropy loss
- No regularization
The weight matrices are initialized with truncated normal (standard deviation of
0.05
) and the biases are initialized to0.05
.Parameters: - batch_size (int) -- Batch size to use.
- weight_decay -- No weight decay (L2-regularization) is used in this
test problem. Defaults to
None
and any input here is ignored.