🚀 Refined BitTransformerLM: Organized codebase with best practices
Browse files
scripts/examples/example.py
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from bit_transformer import example_training_step
|
| 2 |
+
|
| 3 |
+
if __name__ == "__main__":
|
| 4 |
+
loss, telemetry = example_training_step()
|
| 5 |
+
print("Training loss:", loss)
|
| 6 |
+
print("Available telemetry:", list(telemetry.keys()))
|