class_docstr: Main model class.
class_docstr_tok: ["Main","model","class","."]
code: def test_step(self, x):
      loss, bpp, mse = self(x, training=False)
      self.loss.update_state(loss)
      self.bpp.update_state(bpp)
      self.mse.update_state(mse)
      return {
          m.name: m.result() for m in [self.loss, self.bpp, self.mse]
      }
  
imports: import argparse
  import functools
  import glob
  import sys
  from absl import app
  from absl.flags import argparse_flags
  import tensorflow
  import tensorflow_compression
  import tensorflow_datasets
label_desc: Use image node_id 4 for calling the MS2020Model obj's underlying member method code with example usage: obj.test_step(x) and returns: unknown