Ejemplo n.º 1
0
    def session(self):
        return self.sessions[-1]


experiment = Experiment()

# Experiment defaults
experiment.name = 'experiment'
experiment.tags = []
experiment.samples = 0
experiment.model = {'fn': None, 'args': [], 'kwargs': {}}
experiment.optimizer = {'fn': None, 'args': [], 'kwargs': {}}
experiment.sessions = []

# Session defaults
session = AutoMunch()
session.losses = {'solubility': 0, 'l1': 0}
session.seed = random.randint(0, 99)
session.cpus = multiprocessing.cpu_count() - 1
session.device = 'cuda' if torch.cuda.is_available() else 'cpu'
session.log = {'when': []}
session.checkpoint = {'when': []}

# Experiment configuration
for string in args.experiment:
    if '=' in string:
        update = parse_dotted(string)
    else:
        with open(string, 'r') as f:
            update = yaml.safe_load(f)
    # If the current session is defined inside the experiment update the session instead
Ejemplo n.º 2
0
from faunadb import query as q
from faunadb.objects import Ref
from faunadb.client import FaunaClient
from dotenv import dotenv_values
from munch import AutoMunch
from pprint import pprint

cfg = AutoMunch(dotenv_values())

# pprint(cfg)
# exit()

fdb = FaunaClient(secret=cfg.FAUNASECRET)

if __name__ == "__main__":
    indexes = fdb.query(q.paginate(q.collections()))
    pprint(indexes)
Ejemplo n.º 3
0
def test_automunch():
    b = AutoMunch()
    b.urmom = {'sez': {'what': 'what'}}
    assert b.urmom.sez.what == 'what'
Ejemplo n.º 4
0
from .dataset import InfectionDataset

parser = ArgumentParser()
parser.add_argument('--model', nargs='+', required=True)
parser.add_argument('--data', nargs='+', required=True, default=[])
parser.add_argument('--options', nargs='+', required=False, default=[])
parser.add_argument('--output', type=str, required=True)

args = parser.parse_args()

# region Collecting phase

# Defaults
model = Munch(fn=None, args=[], kwargs={}, state_dict=None)
data = []
options = AutoMunch()
options.cpus = multiprocessing.cpu_count() - 1
options.device = 'cuda' if torch.cuda.is_available() else 'cpu'
options.output = args.output

# Model from --model args
for string in args.model:
    if '=' in string:
        update = parse_dotted(string)
    else:
        with open(string, 'r') as f:
            update = yaml.safe_load(f)
            # If the yaml file contains an entry with key `model` use that one instead
            if 'model' in update.keys():
                update = update['model']
    update_rec(model, update)
Ejemplo n.º 5
0
def test_automunch():
    b = AutoMunch()
    b.urmom = {'sez': {'what': 'what'}}
    assert b.urmom.sez.what == 'what'
Ejemplo n.º 6
0
def test_automunch():
    b = AutoMunch()
    b.urmom = {'sez': {'what': 'what'}}
    assert b.urmom.sez.what == 'what'  # pylint: disable=no-member