Commit
•
4f5251a
1
Parent(s):
26fcf3a
Update audioldm_train/modules/audiomae/AudioMAE.py
Browse files
audioldm_train/modules/audiomae/AudioMAE.py
CHANGED
|
@@ -5,8 +5,8 @@ Reference Repo: https://github.com/facebookresearch/AudioMAE
|
|
| 5 |
import torch
|
| 6 |
import torch.nn as nn
|
| 7 |
from timm.models.layers import to_2tuple
|
| 8 |
-
import audioldm_train.modules.audiomae.models_vit as models_vit
|
| 9 |
-
import audioldm_train.modules.audiomae.models_mae as models_mae
|
| 10 |
|
| 11 |
# model = mae_vit_base_patch16(in_chans=1, audio_exp=True, img_size=(1024, 128))
|
| 12 |
|
|
|
|
| 5 |
import torch
|
| 6 |
import torch.nn as nn
|
| 7 |
from timm.models.layers import to_2tuple
|
| 8 |
+
import qa_mdt.audioldm_train.modules.audiomae.models_vit as models_vit
|
| 9 |
+
import qa_mdt.audioldm_train.modules.audiomae.models_mae as models_mae
|
| 10 |
|
| 11 |
# model = mae_vit_base_patch16(in_chans=1, audio_exp=True, img_size=(1024, 128))
|
| 12 |
|