How do I convert a tensorflow model to pytorch?

YuhskeHujisaki · November 15, 2022, 7:33am

IoBT-VISTEC/MIN2Net/blob/main/min2net/model/MIN2Net.py

from tensorflow.keras.layers import Concatenate, AveragePooling2D, BatchNormalization, Conv2D, Conv2DTranspose, Dense, Input, Reshape, Flatten
from tensorflow.keras.constraints import max_norm
from tensorflow.keras.models import Model
from tensorflow.keras.callbacks import CSVLogger, ModelCheckpoint, ReduceLROnPlateau
from tensorflow.keras.callbacks import EarlyStopping
from tensorflow.keras.optimizers import Adam
import tensorflow as tf
from tensorflow.keras import backend as K

import numpy as np
import os
import time
from sklearn.metrics import classification_report, f1_score
from min2net.loss import mean_squared_error, triplet_loss, SparseCategoricalCrossentropy
from min2net.utils import TimeHistory, compute_class_weight

class MIN2Net:
    def __init__(self,
                input_shape=(1,400,20), 
                num_class=2,

This file has been truncated. show original

I want to write MIN2Net with pytorch, how should I convert it?

khushi-411 · November 15, 2022, 11:07am

Hi, You can check out this tutorial for an example!

Similar question (posting for reference that will help you): How to convert this code to pytorch format.

Feel free to ping me if you need further help or are stuck in error. Thanks!