github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fixed CI dataset directory; fixed import error of 2.5d accuracy (#255)

pull/394/head
アマデウス 2022-02-24 14:33:45 +08:00 committed by Frank Lee
parent 219df6e685
commit e13293bb4c
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/PR_CI.yml vendored
View File

@ -28,7 +28,7 @@ jobs:
runs-on: [self-hosted, gpu] runs-on: [self-hosted, gpu]
container: container:
image: nvcr.io/nvidia/pytorch:21.07-py3 image: nvcr.io/nvidia/pytorch:21.07-py3
options: --gpus all --rm --ipc=host -v /data/cifar-10:/data/cifar-10 options: --gpus all --rm --ipc=host -v /data/scratch/cifar-10:/data/scratch/cifar-10
timeout-minutes: 20 timeout-minutes: 20
steps: steps:
- name: Setup Environment - name: Setup Environment
@ -48,7 +48,7 @@ jobs:
run: | run: |
pytest tests pytest tests
env: env:
DATA: /data/cifar-10 DATA: /data/scratch/cifar-10
format_check: format_check:
name: Format Check name: Format Check

2
colossalai/nn/metric/accuracy_2p5d.py
View File

@ -1,5 +1,5 @@
import torch import torch
from colossalai.nn.layer.parallel_2p5d import reduce_by_batch_2p5d from colossalai.nn.layer.parallel_2p5d import reduce_by_batch_2p5d, split_tensor_2p5d
from torch import nn from torch import nn
from ._utils import calc_acc from ._utils import calc_acc