Verified Commit 50909bed authored by Jakob Moser's avatar Jakob Moser
Browse files

Add logging for train method start

parent 8d00141c
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -23,4 +23,5 @@ def train(
    """
    Pre-train a Transformer model from scratch and store the weights on disk.
    """
    logging.info(f"Starting training for {model}, configuration {query_equals_key}, seed {seed}")
    pass  # TODO Implement