vis_sandbox.py [657:662]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            if FLAGS.model == "transformer":
                model = RotomerTransformerModel(FLAGS).eval()
            elif FLAGS.model == "fc":
                model = RotomerFCModel(FLAGS).eval()
            elif FLAGS.model == "s2s":
                model = RotomerSet2SetModel(FLAGS).eval()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



vis_sandbox.py [669:674]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if FLAGS.model == "transformer":
            model = RotomerTransformerModel(FLAGS).eval()
        elif FLAGS.model == "fc":
            model = RotomerFCModel(FLAGS).eval()
        elif FLAGS.model == "s2s":
            model = RotomerSet2SetModel(FLAGS).eval()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -