Exemplos de get_num_augmentations em Python

Exemplo n.º 1

Arquivo: 06_train-flat-singletask-coarse.py Projeto: aascode/cramer2020icassp

aug_kind_str = args.aug_kind_str
trial_str = args.trial_str
lr = args.lr
base_wd = args.base_wd
batch_size = args.batch_size
epochs = args.epochs
tfr_str = args.tfr
num_cpus = args.num_cpus
align_perturb = args.align_perturb
lr_annealing = args.lr_annealing

active_streamers = 64
streamer_rate = 1024
steps_per_epoch = int(
    np.ceil(36.0 * localmodule.get_num_augmentations(aug_kind_str) *
            streamer_rate / batch_size))

# Iterate over the entire training validation set
valid_batch_size = 512
validation_steps = int(np.ceil(35335 / float(valid_batch_size)))

# Set number of epochs.
if not epochs:
    if aug_kind_str == "none":
        epochs = 512
    else:
        epochs = 1024

# Print header.
start_time = int(time.time())

Exemplo n.º 2

aug_kind_str = args.aug_kind_str
trial_str = args.trial_str
lr = args.lr
base_wd = args.base_wd
batch_size = args.batch_size
epochs = args.epochs
tfr_str = args.tfr
num_cpus = args.num_cpus
align_perturb = args.align_perturb
lr_annealing = args.lr_annealing


active_streamers = 64
streamer_rate = 1024
steps_per_epoch = int(np.ceil(36.0 * localmodule.get_num_augmentations(aug_kind_str) * streamer_rate / batch_size))

# Iterate over the entire training validation set
valid_batch_size = 512
validation_steps = int(np.ceil(35335 / float(valid_batch_size)))


# Set number of epochs.
if not epochs:
    if aug_kind_str == "none":
        epochs = 512
    else:
        epochs = 1024


# Print header.