amzn / amzn/trans-encoder

Run code without CUDA

Open
#9 1 comment 1 reaction 0 assignees View on GitHub
Dominant language
Python
Stars
133
Forks
16
PR merge metrics
No merged PRs in 30d

Description

First of all, great paper and great that you guys make your code publish available. :-)

For both I need CUDA: train_self_distill.sh and train_mutual_distill.sh. Does the repo/training also works without CUDA available?

Contributor guide

Open the contributing guide

Research direction

Start by reviewing train_self_distill.sh and train_mutual_distill.sh to identify where CUDA is required, then try the repository's training flow in an environment without CUDA. Done means both training scripts can run without CUDA, or the supported limitation is documented clearly.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.