Closed harbecke closed 4 years ago
should solve #16 training_data and validation_data are created separately, this lead to the following changes
training_data
validation_data
hex.training.train.train
hex.creation.create_data.create_self_play_data
num_samples
args
verbose
torch.save()
config
train_samples_per_model
val_samples_per_model
[CREATE DATA]
samples_per_model
num_validation_data
[TRAIN]
👍
should solve #16
training_data
andvalidation_data
are created separately, this lead to the following changeshex.training.train.train
now takestraining_data
andvalidation_data
separatelyhex.creation.create_data.create_self_play_data
takesnum_samples
as own argument and fromargs
anymore and has averbose
flag (for validation data)torch.save()
saves both data in the same file (this can be changed quite easily, if you want)config
now hastrain_samples_per_model
andval_samples_per_model
in[CREATE DATA]
instead ofsamples_per_model
andnum_validation_data
in[TRAIN]