fix

Author: moehanabi

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):