↓ 1 callersFunctiontrain(latent_dim, learning_rate, number_filters, vae_alpha, vae_beta, kl_divergence_weight)
accelerate/accelerate_logcosh.py:93
↓ 1 callersFunctionupfirdn2d_native(
input, kernel, up_x, up_y, down_x, down_y, pad_x0, pad_x1, pad_y0, pad_y1
)
op/upfirdn2d.py:159
Method__init__(
self, model="net-lin", net="alex", colorspace="rgb", spatial=False, use_gpu=True, gpu_ids=[0]
)
validation/lpips/__init__.py:14