Skip to content

[bugfix] fix correctness of kjt.lengths when ShardedEmbeddingBag’s pooling_type is mean and shard_type is row_wise #347

[bugfix] fix correctness of kjt.lengths when ShardedEmbeddingBag’s pooling_type is mean and shard_type is row_wise

[bugfix] fix correctness of kjt.lengths when ShardedEmbeddingBag’s pooling_type is mean and shard_type is row_wise #347

Workflow file for this run

name: Unit Test CI
on:
pull_request:
types: [opened, reopened, synchronize]
workflow_dispatch:
jobs:
ci-test:
runs-on: tzrec-runner
container:
image: mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easyrec/tzrec-devel:0.7
options: --gpus all --ipc host
steps:
- name: FetchCommit ${{ github.event.pull_request.head.sha }}
uses: actions/checkout@v2
with:
ref: ${{ github.event.pull_request.head.sha }}
path: run_${{ github.run_id }}
- name: RunUnitTestCI
id: run_unittest_ci
run: |
cd run_${{ github.run_id }}
bash scripts/ci_test.sh