Browse Source

Update attn.py

main
BBing 2 years ago
parent
commit
b5545a7f23
1 changed files with 0 additions and 5 deletions
  1. +0
    -5
      models/attn.py

+ 0
- 5
models/attn.py View File

@@ -1,9 +1,4 @@

import numpy as np

from math import sqrt
from utils.masking import TriangularCausalMask, ProbMask

import numpy as np
import mindspore.nn as nn
import mindspore.ops as ops


Loading…
Cancel
Save