From 4e926318265c85411f10e023bb11f9a22f1c248a Mon Sep 17 00:00:00 2001 From: Mateusz Kopcinski Date: Tue, 20 Aug 2024 10:53:57 +0200 Subject: [PATCH] added typedoc --- lib/ex_vision/style_transfer/style_transfer.ex | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/lib/ex_vision/style_transfer/style_transfer.ex b/lib/ex_vision/style_transfer/style_transfer.ex index b17fa9c..b74f3f0 100644 --- a/lib/ex_vision/style_transfer/style_transfer.ex +++ b/lib/ex_vision/style_transfer/style_transfer.ex @@ -23,6 +23,10 @@ for {module, opts} <- Configuration.configuration() do #{module} is a custom style transfer model optimised for devices with low computational capabilities and CPU inference. """ require Logger + + @typedoc """ + A type consisting of output tesnor (stylized image tensor) from style transfer models of shape {#{Enum.join(Tuple.to_list(opts[:resolution]) ++ [3], ", ")}}. + """ @type output_t() :: Nx.Tensor.t() use ExVision.Model.Definition.Ortex, model: unquote(opts[:model])