File size: 520 Bytes
15063d0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
"""
Mock framework module for ndr geometric attention
只保留必要的部分
"""
import torch
from typing import Optional, Any
class visualize:
"""Mock visualize class"""
@staticmethod
def attention(*args, **kwargs):
"""Dummy attention visualization"""
pass
@staticmethod
def plot(*args, **kwargs):
"""Dummy plot"""
pass
# Mock其他可能需要的功能
def get_logger(name: str):
"""Mock logger"""
import logging
return logging.getLogger(name)
|