Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Model] Implement DR-GST #161

Merged
merged 19 commits into from
Nov 30, 2023
Merged

[Model] Implement DR-GST #161

merged 19 commits into from
Nov 30, 2023

Conversation

R0k1e
Copy link
Contributor

@R0k1e R0k1e commented Aug 2, 2023

Description

the executable version of DR-GST, only support torch backend

Checklist

Please feel free to remove inapplicable items for your PR.

  • The PR title starts with [$CATEGORY] (such as [NN], [Model], [Doc], [Feature]])
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • Code is well-documented
  • To the best of my knowledge, examples are either not affected by this change,
    or have been fixed to be compatible with this change
  • Related issue is referred in this PR

Changes

DR-GST.py Outdated
train_index = mask_to_index(idx_train)
idx_train_ag = mask_to_index(data.train_mask)
if tlx.BACKEND == 'torch':
pseudo_labels = labels.clone() # TODO tensorflow 会报错
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think you may use the copy.deepcopy() to implement this function.

@dddg617 dddg617 changed the title [Model] DR-GST torch version [Model] Implement DR-GST Nov 29, 2023
@dddg617 dddg617 merged commit 9f7e62a into BUPT-GAMMA:main Nov 30, 2023
dddg617 added a commit that referenced this pull request Jan 12, 2024
* DR-GST torch版本

* 整合进example与model,添加未补充结果的readme

* 除Flickr以外数据

* 除Flickr以外数据

* 除Flickr以外数据

* 修正backward bug

* 修正本地dataset

* 修正命名

* 添加train1

* 添加train1

* 添加train1

* 添加train1

* 修正bald

* DR_GST README

* nightly update

* [Model] Fix DR_GST

* [Doc] Update README

* [Doc] Update readme

---------

Co-authored-by: dddg617 <[email protected]>
Co-authored-by: dddg617 <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants