from numpy.random import random
from labml import tracker, experiment as exp

conf = {'batch_size': 20}

def train(n:int):
    loss = 0.999 ** n + random() / 10
    accuracy = 1 - 0.999 ** n + random() / 10
    return loss, accuracy

with exp.record(name='sample', exp_conf=conf):
    for i in range(100000):
        lss, acc = train(i)
        tracker.save(i, loss=lss, accuracy=acc)
                
            

                from labml import experiment as exp
from labml.utils.lightning import LabMLLightningLogger 

trainer = pl.Trainer(gpus=1,
                     max_epochs=5,
                     progress_bar_refresh_rate=20,
                     logger=LabMLLightningLogger())

with exp.record(name='sample', exp_conf=conf):
        trainer.fit(model, data_loader)
                
            

                from labml import experiment as exp
from labml.utils.keras import LabMLKerasCallback

with exp.record(name='sample', exp_conf=conf):
    for i in range(50):
        model.fit(x_train,
                  y_train,
                  epochs=conf['epochs'],
                  validation_data=(x_test, y_test),
                  callbacks=[LabMLKerasCallback()],
                  verbose=None)