WCNegentropy's picture
๐Ÿš€ Refined BitTransformerLM: Organized codebase with best practices
3928bd8 verified
raw
history blame contribute delete
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()))