Skip to content

GPU versions of example 3 to 6 #5

@esvhd

Description

@esvhd

Hi @vinhkhuc ,

Would you be interested in a PR for the cuda version of these 4 models? I made some simple modifications to make them work on GPUs, while trying to preserve the flow of the original non-GPU versions.

Example 1 & 2 probably doesn't need a GPU version since they train quite fast anyway.

Thx

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions