Commit 18ba760b authored by Cianciosa, Mark's avatar Cianciosa, Mark
Browse files

Fix comment to show the correct model.

parent 671be670
Loading
Loading
Loading
Loading
−3.83 KiB (327 KiB)

File changed.

No diff preview for this file type.

+1 −1
Original line number Diff line number Diff line
@@ -58,7 +58,7 @@
      END SUBROUTINE

!-------------------------------------------------------------------------------
!>  @brief Test a keras model.
!>  @brief Test a custom pytorch model.
!-------------------------------------------------------------------------------
      SUBROUTINE test_custom_pytorch
      USE, INTRINSIC :: ISO_C_BINDING
+1 −1
Original line number Diff line number Diff line
@@ -6,7 +6,7 @@
#include <ml_embedder.hpp>

//------------------------------------------------------------------------------
///  @brief Test a keras model.
///  @brief Test a custom pytorch model.
//------------------------------------------------------------------------------
void test_custom_pytorch() {
    ml_embedder::custom_pytorch model(TGLF_MODEL_WEIGHTS, TGLF_MODEL_CONFIG);