Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
B
BTS-MTGNN
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
zhlj
BTS-MTGNN
Commits
ad969d47
Commit
ad969d47
authored
Jan 31, 2025
by
zlj
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
delete tt
parent
673bea9c
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
examples/test_all.sh
+3
-3
No files found.
examples/test_all.sh
View file @
ad969d47
...
...
@@ -10,7 +10,7 @@ partitions="8"
node_per
=
"4"
nnodes
=
"2"
node_rank
=
"0"
probability_params
=(
"0.
1
"
)
probability_params
=(
"0.
2"
"0.4"
"0.6"
"0.8
"
)
sample_type_params
=(
"boundery_recent_decay"
)
#"boundery_recent_decay")
#sample_type_params=("recent" "boundery_recent_decay") #"boundery_recent_uniform")
...
...
@@ -80,13 +80,13 @@ for data in "${data_param[@]}"; do
if
[
"
$mem
"
=
"historical"
]
;
then
for
ssim
in
"
${
shared_memory_ssim
[@]
}
"
;
do
if
[
"
$partition
"
=
"ours"
]
||
[
"
$partition
"
=
"metis"
]
;
then
torchrun
--nnodes
"
$nnodes
"
--node_rank
"
$node_rank
"
--nproc-per-node
"
$node_per
"
--master-addr
"
$addr
"
--master-port
9445 train_boundery.py
--dataname
"
$data
"
--mode
"
$model
"
--partition
"
$partition
"
--topk
"
$topk
"
--sample_type
"
$sample
"
--probability
"
$pro
"
--memory_type
"
$mem
"
--shared_memory_ssim
"
$ssim
"
--seed
"
$seed
"
--neighbor
"
$neighbor
"
>
all_
"
$seed
"
/
"
$data
"
/
"
$model
"
/
"
$partitions
"
-
"
$partition
"
-
"
$topk
"
-
"
$mem
"
-
"
$ssim
"
-
"
$sample
"
-
-
$pro
"-"
$neighbor
".out &
torchrun
--nnodes
"
$nnodes
"
--node_rank
"
$node_rank
"
--nproc-per-node
"
$node_per
"
--master-addr
"
$addr
"
--master-port
9445 train_boundery.py
--dataname
"
$data
"
--mode
"
$model
"
--partition
"
$partition
"
--topk
"
$topk
"
--sample_type
"
$sample
"
--probability
"
$pro
"
--memory_type
"
$mem
"
--shared_memory_ssim
"
$ssim
"
--seed
"
$seed
"
--neighbor
"
$neighbor
"
>
all_
"
$seed
"
/
"
$data
"
/
"
$model
"
/
"
$partitions
"
-
"
$partition
"
-
"
$topk
"
-
"
$mem
"
-
"
$ssim
"
-
"
$sample
"
-
"
$pro
"
-
"
$neighbor
"
.out &
wait
fi
done
elif
[
"
$mem
"
=
"all_reduce"
]
;
then
if
[
"
$partition
"
=
"ours"
]
;
then
torchrun --nnodes "
$nnodes
" --node_rank "
$node_rank
" --nproc-per-node "
$node_per
" --master-addr "
$addr
" --master-port 9445 train_boundery.py --dataname "
$data
" --mode "
$model
" --partition "
$partition
" --topk "
$topk
" --sample_type "
$sample
" --probability "
$pro
" --memory_type "
$mem
" --seed "
$seed
" --neighbor "
$neighbor
" > all_"
$seed
"/"
$data
"/"
$model
"/"
$partitions
"-"
$partition
"-"
$topk
"-"
$mem
"-"
$ssim
"-"
$sample
"-
-
$pro
"
-
"
$neighbor
"
.out &
torchrun
--nnodes
"
$nnodes
"
--node_rank
"
$node_rank
"
--nproc-per-node
"
$node_per
"
--master-addr
"
$addr
"
--master-port
9445 train_boundery.py
--dataname
"
$data
"
--mode
"
$model
"
--partition
"
$partition
"
--topk
"
$topk
"
--sample_type
"
$sample
"
--probability
"
$pro
"
--memory_type
"
$mem
"
--seed
"
$seed
"
--neighbor
"
$neighbor
"
>
all_
"
$seed
"
/
"
$data
"
/
"
$model
"
/
"
$partitions
"
-
"
$partition
"
-
"
$topk
"
-
"
$mem
"
-
"
$ssim
"
-
"
$sample
"
-
"
$pro
"
-
"
$neighbor
"
.out &
wait
fi
else
...
...
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