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
0a01432368
commit
f123ca8a9b
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      utils/masking.py

+ 1
- 1
utils/masking.py View File

@@ -1,5 +1,5 @@
import torch
4
5
class TriangularCausalMask():
def __init__(self, B, L, device="cpu"):
mask_shape = [B, 1, L, L]


Write Preview
Loading…
Cancel
Save