Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
fairscale
Commits
8f8f8ef9
Unverified
Commit
8f8f8ef9
authored
Sep 23, 2022
by
Min Xu
Committed by
GitHub
Sep 23, 2022
Browse files
[chore] move fair_dev into fairscale (#1078)
Co-authored-by:
Min Xu
<
min.xu.public@gmail.com
>
parent
bfd57ff3
Changes
63
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
6 additions
and
6 deletions
+6
-6
tests/optim/test_oss_adascale.py
tests/optim/test_oss_adascale.py
+2
-2
tests/optim/test_single_node_adascale.py
tests/optim/test_single_node_adascale.py
+3
-3
tests/utils/test_reduce_scatter_bucketer.py
tests/utils/test_reduce_scatter_bucketer.py
+1
-1
No files found.
tests/optim/test_oss_adascale.py
View file @
8f8f8ef9
...
@@ -22,8 +22,8 @@ from torch.nn import Linear, Sequential
...
@@ -22,8 +22,8 @@ from torch.nn import Linear, Sequential
from
torch.nn.parallel
import
DistributedDataParallel
as
DDP
from
torch.nn.parallel
import
DistributedDataParallel
as
DDP
from
torch.optim
import
SGD
from
torch.optim
import
SGD
from
fair_dev.testing.golden_testing_data
import
adascale_test_data
from
fairscale.
fair_dev.testing.golden_testing_data
import
adascale_test_data
from
fair_dev.testing.testing
import
skip_if_single_gpu
from
fairscale.
fair_dev.testing.testing
import
skip_if_single_gpu
from
fairscale.optim
import
OSS
,
AdaScale
,
AdaScaleWrapper
from
fairscale.optim
import
OSS
,
AdaScale
,
AdaScaleWrapper
...
...
tests/optim/test_single_node_adascale.py
View file @
8f8f8ef9
...
@@ -19,9 +19,9 @@ from torch.nn import Linear, Sequential
...
@@ -19,9 +19,9 @@ from torch.nn import Linear, Sequential
from
torch.optim
import
SGD
from
torch.optim
import
SGD
from
torch.optim.lr_scheduler
import
LambdaLR
from
torch.optim.lr_scheduler
import
LambdaLR
from
fair_dev.testing.golden_testing_data
import
adascale_test_data
from
fairscale.
fair_dev.testing.golden_testing_data
import
adascale_test_data
from
fair_dev.testing.testing
import
make_cudnn_deterministic
,
skip_if_no_cuda
from
fairscale.
fair_dev.testing.testing
import
make_cudnn_deterministic
,
skip_if_no_cuda
from
fair_dev.testing.testing_memory
import
find_tensor_by_shape
from
fairscale.
fair_dev.testing.testing_memory
import
find_tensor_by_shape
from
fairscale.optim
import
AdaScale
from
fairscale.optim
import
AdaScale
...
...
tests/utils/test_reduce_scatter_bucketer.py
View file @
8f8f8ef9
...
@@ -12,7 +12,7 @@ from unittest import mock
...
@@ -12,7 +12,7 @@ from unittest import mock
from
parameterized
import
parameterized
from
parameterized
import
parameterized
import
torch
import
torch
from
fair_dev.testing.testing
import
dist_init
,
spawn_for_all_world_sizes
from
fairscale.
fair_dev.testing.testing
import
dist_init
,
spawn_for_all_world_sizes
from
fairscale.internal
import
torch_version
from
fairscale.internal
import
torch_version
from
fairscale.internal.reduce_scatter_bucketer
import
ReduceScatterBucketer
from
fairscale.internal.reduce_scatter_bucketer
import
ReduceScatterBucketer
...
...
Prev
1
2
3
4
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment