9 lines
201 B
Python
9 lines
201 B
Python
|
############### Configuration file for Training of SplitMNIST and Mixtures ###############
|
||
|
|
n_epochs = 5
|
||
|
|
lr_start = 0.001
|
||
|
|
num_workers = 4
|
||
|
|
valid_size = 0.2
|
||
|
|
batch_size = 256
|
||
|
|
train_ens = 15
|
||
|
|
valid_ens = 10
|