BP1 - 02

BP1 - 02.py

class BatchCounter(Callback): 
    def begin_epoch(self, epoch):
        self.epoch=epoch
        self.batch_counter = 1
        return True
        
    def after_step(self):
        self.batch_counter += 1
        if self.batch_counter % 200 == 0: print(f'Batch {self.batch_counter} completed')
        return True
        
class TimeCheck(Callback):
    def begin_fit(self, learn):
        self.learn = learn
        self.epoch_counter = 1
        return True
    
    def begin_epoch(self, epoch):
        self.epoch=epoch
        print(f'Epoch {self.epoch_counter} started at {time.strftime("%H:%M:%S", time.gmtime())}')
        self.epoch_counter += 1
        return True 
        
 class PrintLoss(Callback):
    def after_epoch(self):
        print(f'Loss: {self.loss}')
        return True