limingjuan
/
Prediction-MindSpore
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

Update masking.py

main
BBing 2 years ago
parent
5c4ab55e86
commit
eb394976a2
1 changed files with 0 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +0
    -1
      utils/masking.py

+ 0
- 1
utils/masking.py View File

@@ -2,7 +2,6 @@ import mindspore
import mindspore.ops.operations as P
import mindspore.numpy as mnp


class TriangularCausalMask():
def __init__(self, B, L, device="cpu"):
mask_shape = [B, 1, L, L]


Write Preview
Loading…
Cancel
Save