fix

tensorlayer · Nov 22, 2024 · 4e94706 · 4e94706
1 parent dcc58c0
commit 4e94706
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/ops/test_ops_torch_nn.py b/tests/ops/test_ops_torch_nn.py
@@ -1,8 +1,9 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 import os
 import unittest
 
+os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
+os.environ['TL_BACKEND'] = 'torch'
+
 import numpy as np
 import torch
 import torch.nn.functional as F
@@ -11,8 +12,6 @@
 from tensorlayerx.backend.ops.torch_nn import *
 from tests.utils import CustomTestCase
 
-os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
-
 
 class TestPaddingFormat(CustomTestCase):