BitTransformerLM / example.py
WCNegentropy's picture
🤖 Updated BitTransformerLM from development space
36c78b1 verified
raw
history blame
216 Bytes
from bit_transformer import example_training_step
if __name__ == "__main__":
loss, telemetry = example_training_step()
print("Training loss:", loss)
print("Available telemetry:", list(telemetry.keys()))