search

neurosim / DNN_NeuroSim_V1.3

Benchmark framework of compute-in-memory based accelerators for deep neural network (inference engine focused)
62 stars 36 forks source link

Creating network.csv files #27

Closed harshari closed 2 years ago

harshari commented 2 years ago

Hello, thank you for NeuroSim. I want to implement some other Network like ResNet50 on ImageNet dataset or VGG19 on CIFAR100. What is the suggestive way of creating the Network.csv file for any other network and dataset we use? Is it getting generated through the pytorch .pth models as we define or we create it manually?

Regards

neurosim commented 2 years ago

Hi! Thanks for using NeuroSim. I created the Network.csv file manually. But you could try to print the layer information (e.g., in_channels, out_channels, kernel_size, stride) from the pytorch model, which might help you to create it.