#!/bin/bash

# cp  /opt/software/github/swav/traced_swav_imagenet_layer2.pt ./
onnxsim /opt/software/github/swav/swav_imagenet_layer2.onnx swav_imagenet_layer2_sim.onnx

# https://github.com/onnx/onnx-tensorflow
onnx-tf convert --infile swav_imagenet_layer2_sim.onnx --outdir tf_saved_model

tensorflowjs_converter --input_format tf_saved_model tf_saved_model tfjs_saved_model/