Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update gnn_layers.py

Browse Source
pull/1578/head
Kai Han 3 years ago committed by GitHub
parent 33c6d8e9a9
commit df75883615
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
timm/models/layers/gnn_layers.py
Unescape View File

@ -4,9 +4,9 @@ import numpy as np
import torch import torch
from torch import nn from torch import nn
import torch.nn.functional as F import torch.nn.functional as F
from timm.models.fx_features import register_notrace_module
from .drop import DropPath from .drop import DropPath
from .pos_embed import build_sincos2d_pos_embed from .pos_embed import build_sincos2d_pos_embed
from .fx_features import register_notrace_module
def pairwise_distance(x, y): def pairwise_distance(x, y):

Write Preview
Loading…
Cancel
Save