microsoft
/
ptgnn
зеркало из https://github.com/microsoft/ptgnn.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность
88 коммитов 6 Ветки 22 Теги 215 KiB
Граф коммитов

88 Коммитов

Автор SHA1 Сообщение Дата
Miltos Allamanis 03ab80499b Explicitly set the starting epoch index to the restored one. 2021-07-22 09:29:11 +01:00
Miltos Allamanis 764a7038ff Fix bugs in distributed saving of optimizer state. 2021-07-22 09:29:11 +01:00
Miltos Allamanis 3bd5966e88 Allow restoring optimizer/scheduler state. 2021-07-22 09:29:11 +01:00
Miltos Allamanis 5020bb7f6f Improve some code comments about message-passing layers. 2021-07-21 09:42:34 +01:00
Miltos 27b3102184
Merge pull request #13 from microsoft/dev/miallama/distributedtrainer 
A multi-GPU distributed trainer
 2021-07-14 14:43:11 +01:00
Miltos Allamanis ce0d8f6814 Fix AMP which was inconsistently implemented: https://pytorch.org/docs/stable/notes/amp_examples.html 2021-07-14 12:40:59 +01:00
Miltos Allamanis 6edb8aeeba Reduce gpu->cpu transfers when progress bar is disabled. 2021-07-14 11:36:08 +00:00
Miltos Allamanis c724767872 Remove unneeded gpu->cpu transfers during metrics update. 2021-07-14 12:04:03 +01:00
Miltos Allamanis cd02705dab Fix how validation-improved hooks work. 2021-07-09 12:08:15 +01:00
Miltos Allamanis 66f0046517 Improved hooks/AML support fixes. 2021-07-09 12:00:11 +01:00
Miltos Allamanis 80e8879469 Reduce insanely long buffering. 2021-07-08 19:16:04 +01:00
Miltos Allamanis ed8aac4bb2 Merge branch 'dev/miallama/distributedtrainer' of github.com:microsoft/ptgnn into dev/miallama/distributedtrainer 2021-07-08 17:23:23 +01:00
Miltos Allamanis 95727b9802 Minor AML fixes. 2021-07-08 17:22:10 +01:00
Miltos Allamanis b22b91fca7 Restore models in-cpu. 2021-07-08 15:45:47 +00:00
Miltos Allamanis 060765d6d6 Remove uncessary `enumerate` 2021-07-08 10:29:10 +01:00
Miltos Allamanis b37f3830c0 Fix typos 2021-07-08 10:28:34 +01:00
Miltos Allamanis 948ba35646 Bypass deadlock by throwing on join() 2021-07-07 22:22:39 +00:00
Miltos Allamanis a3a841676a Repair bug. 2021-07-07 13:55:33 +00:00
Miltos Allamanis 25e9b29e28 Polish distributed trainer. 2021-07-07 08:03:23 +00:00
Miltos Allamanis 12496e4ee4 WIP implementation of distributed data parallel training. 2021-07-06 22:26:59 +00:00
Miltos 8df6d85368
Merge pull request #12 from microsoft/dev/miallam/ecg-implementation 
An implementation of ECG
 2021-04-16 08:05:45 +01:00
Miltos Allamanis 5e7e146cc0 An implementation of ECG 2021-04-12 13:54:48 +01:00
Miltos d30cb0d585
Merge pull request #10 from microsoft/dev/miallama/better-names 
Add more informative aliases for the GnnOutput
 2021-03-16 14:24:30 +00:00
Miltos Allamanis b36fd52f37 Add more informative aliases for the GnnOutput 2021-03-16 13:46:33 +00:00
Miltos Allamanis 9f0c0f0019 Fix AMP issue in PNA. 2021-03-12 21:13:57 +00:00
Miltos 207074d246
Merge pull request #8 from microsoft/dev/miallama/pna-layer 
Add PNA layer.
 2021-03-12 07:55:42 +00:00
Miltos Allamanis 6a00b4a65c Simplify PNA aggregation/layers. 2021-03-11 17:12:06 +00:00
Miltos Allamanis 4d45e10a4f Merge branch 'master' into dev/miallama/pna-layer 2021-03-11 16:21:22 +00:00
Miltos Allamanis 35b45d5f8b Update AML files. 2021-03-11 16:20:50 +00:00
Miltos Allamanis e627ca507d Update AML files. 2021-03-11 16:19:29 +00:00
Miltos 66cf9eae29
Update setup.py 2021-02-23 12:47:01 +00:00
Miltos Allamanis 7f59fb0a84 Merge branch 'master' into dev/miallama/pna-layer 2021-02-22 10:15:47 +00:00
Miltos 7a23ce4910
Merge pull request #9 from microsoft/dev/miallama/edge-features 
Allow using arbitrary edge features in GNNs
 2021-02-22 10:12:23 +00:00
Miltos Allamanis 8dca26f8cc Address Marc's feedback. 2021-02-22 09:47:35 +00:00
Miltos Allamanis b42df7765d Bug fix on no edge features. 2021-02-15 12:57:02 +00:00
Miltos Allamanis e5e1aff090 Fixes to the edge-attrs. 2021-02-15 12:41:49 +00:00
Miltos Allamanis d5702cda39 Allow adding features to edges. 2021-02-15 12:16:01 +00:00
Miltos Allamanis bc5323d2fd Minor AML updates. 2021-01-25 08:32:49 +00:00
Miltos Allamanis b4525bec3d Add PNA layer. 2021-01-12 14:59:34 +00:00
Miltos 16c7ce6c7d
Update pre-commit.yml 2020-12-07 18:12:37 +00:00
Miltos Allamanis b2bbd7d2b5 Update how torch-scatter is loaded, given upstream changes. 2020-12-07 17:47:20 +00:00
Miltos Allamanis f0340bc47a Always save after metadata loading. 2020-12-07 17:41:30 +00:00
Miltos 2ef1159b4b
Merge pull request #7 from microsoft/mallamanis/simplify-saving 
Simplify model saving, when parent folder doesn't exist.
 2020-11-27 14:15:17 +00:00
Miltos Allamanis cceac45eb3 Add import... 2020-11-27 12:35:20 +00:00
Miltos Allamanis ab3e3091d7 Simplify saving, when parent folder doesn't exist. 2020-11-27 12:29:50 +00:00
Miltos 0ba830866e
Merge pull request #6 from microsoft/dev/mallamanis/amp 
Support AMP training in GNNs and other neural models.
 2020-11-10 14:53:17 +00:00
Miltos Allamanis 3bfe1f8f7a Update test matrix to allow support for AMP 2020-11-10 14:13:39 +00:00
Miltos Allamanis f12abb4c01 Support AMP training in GNNs and other neural models. 2020-11-10 14:04:56 +00:00
Miltos e1c507f399
Merge pull request #5 from microsoft/dev/mallamanis/expose-shuffling-option 
Expose option to shuffle training tensor iterator.
 2020-09-29 07:45:11 +01:00
Miltos Allamanis 7d19a30ad4 Expose option to shuffle training tensor iterator. 2020-09-28 23:33:50 +01:00