wsntxxn
commited on
Commit
•
def3c02
1
Parent(s):
6065472
Update requirements.txt
Browse files- models/cnn_encoder.py +0 -2
- requirements.txt +3 -1
models/cnn_encoder.py
CHANGED
|
@@ -1,7 +1,5 @@
|
|
| 1 |
# -*- coding: utf-8 -*-
|
| 2 |
|
| 3 |
-
from einops import rearrange
|
| 4 |
-
|
| 5 |
import torch
|
| 6 |
import torch.nn as nn
|
| 7 |
import torch.nn.functional as F
|
|
|
|
| 1 |
# -*- coding: utf-8 -*-
|
| 2 |
|
|
|
|
|
|
|
| 3 |
import torch
|
| 4 |
import torch.nn as nn
|
| 5 |
import torch.nn.functional as F
|
requirements.txt
CHANGED
|
@@ -1,2 +1,4 @@
|
|
| 1 |
efficientnet_pytorch
|
| 2 |
-
PyYAML
|
|
|
|
|
|
|
|
|
| 1 |
efficientnet_pytorch
|
| 2 |
+
PyYAML
|
| 3 |
+
torchaudio
|
| 4 |
+
einops
|