update to core_v0.9

4b097dee · liangjing · 3aca1415 · 4b097dee · 4b097dee · 4b097dee
Commit 4b097dee authored Oct 29, 2024 by liangjing
20 changed files
--- a/megatron/model/vision/knn_monitor.py
+++ b/megatron/model/vision/knn_monitor.py
--- a/megatron/model/vision/mit_backbone.py
+++ b/megatron/model/vision/mit_backbone.py
--- a/megatron/model/vision/swin_backbone.py
+++ b/megatron/model/vision/swin_backbone.py
--- a/megatron/model/vision/utils.py
+++ b/megatron/model/vision/utils.py
--- a/megatron/model/vision/vit_backbone.py
+++ b/megatron/model/vision/vit_backbone.py
--- a/megatron/legacy/mpu/tests/__init__.py
+++ b/megatron/legacy/mpu/tests/__init__.py
--- a/megatron/mpu/tests/commons.py
+++ b/megatron/mpu/tests/commons.py
--- a/megatron/mpu/tests/test_cross_entropy.py
+++ b/megatron/mpu/tests/test_cross_entropy.py
--- a/megatron/mpu/tests/test_data.py
+++ b/megatron/mpu/tests/test_data.py
--- a/megatron/mpu/tests/test_initialize.py
+++ b/megatron/mpu/tests/test_initialize.py
--- a/megatron/mpu/tests/test_layers.py
+++ b/megatron/mpu/tests/test_layers.py
--- a/megatron/mpu/tests/test_random.py
+++ b/megatron/mpu/tests/test_random.py
--- a/megatron/memory.py
+++ b/megatron/memory.py
--- a/megatron/microbatches.py
+++ b/megatron/microbatches.py
--- a/megatron/model/distributed.py
+++ b/megatron/model/distributed.py
--- a/megatron/optimizer/__init__.py
+++ b/megatron/optimizer/__init__.py
--- a/megatron/optimizer/distrib_optimizer.py
+++ b/megatron/optimizer/distrib_optimizer.py
--- a/megatron/__init__.py
+++ b/megatron/__init__.py
--- a/megatron/training/activations.py
+++ b/megatron/training/activations.py
--- a/megatron/arguments.py
+++ b/megatron/arguments.py