You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
from torch import nn
|
|
|
|
from imaginairy.vendored.refiners.fluxion.layers.module import Module
|
|
|
|
|
|
class ReflectionPad2d(nn.ReflectionPad2d, Module):
|
|
def __init__(self, padding: int) -> None:
|
|
super().__init__(padding=padding)
|