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
282ce9a1
Commit
282ce9a1
authored
Feb 20, 2025
by
zlj
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix train_remote_pos and train_remote_neg delay
parent
df36000d
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 additions
and
1 deletions
+1
-1
starrygl/module/utils.py
+1
-1
No files found.
starrygl/module/utils.py
View file @
282ce9a1
...
...
@@ -159,7 +159,7 @@ class AdaParameter:
self
.
beta
=
self
.
beta
*
average_gnn_aggregate
/
average_fetch
*
(
1
+
self
.
wait_threshold
)
average_memory_sync_time
=
self
.
average_memory_sync
/
self
.
count_memory_sync
average_memory_update_time
=
self
.
average_memory_update
/
self
.
count_memory_update
self
.
alpha
=
(
2
-
math
.
pow
((
2
-
self
.
alpha
),
average_memory_update_time
/
average_memory_sync_time
*
(
1
+
self
.
wait_threshold
)
))
self
.
alpha
=
self
.
alpha
+
math
.
log
(
average_memory_update_time
/
average_memory_sync_time
*
(
1
+
self
.
wait_threshold
))
self
.
beta
=
max
(
min
(
self
.
beta
,
self
.
max_beta
),
self
.
min_beta
)
self
.
alpha
=
max
(
min
(
self
.
alpha
,
self
.
max_alpha
),
self
.
min_alpha
)
ctx
=
DistributedContext
.
get_default_context
()
...
...
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