Commit 64bae470 authored by CJ's avatar CJ
Browse files

Fix import partial

parent 80ad8ccf
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
import math
from functools import wraps
from functools import wraps, partial

import torch
import torch.nn.functional as F