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 numpy as np
import mindspore.nn as nn import mindspore.nn as nn
import mindspore.ops as ops import mindspore.ops as ops


Loading…
Cancel
Save