示例#1
0
def default_main(seq_trial_cb, non_repeating=True):
    if non_repeating:
        seq_trial_cb = NonRepeatingTrials(seq_trial_cb)
    try:
        batch_util.seq_trials_main(seq_trial_cb, State)
    except batch_util.StopBatch as e:
        assert e.error
        sys.exit(1)
示例#2
0
def default_main(seq_trial_cb, non_repeating=True):
    _patch_numpy_deprecation_warnings()
    if non_repeating:
        seq_trial_cb = NonRepeatingTrials(seq_trial_cb)
    batch_util.seq_trials_main(State, seq_trial_cb)
示例#3
0
import sys

from guild import batch_util

batch_fail = False
trials_fail = ""


def init_state(batch):
    return batch_util.SeqState(batch, trial=0, trials_fail=_trials_fail())


def _trials_fail():
    return [int(s) for s in str(trials_fail).split(",") if s]


def init_trial(trial, state):
    flags = {}
    flags.update(state.proto_flags)
    if state.trial in state.trials_fail:
        if batch_fail:
            print("BATCH FAIL")
            sys.exit(1)
        flags["fail"] = True
    state.trial += 1
    return flags, {}


batch_util.seq_trials_main(init_trial, init_state)