Skip to content

modify resnet model to use load dataset#429

Open
ctr-lelanchelianTT wants to merge 2 commits intomainfrom
lelanchelian/check_resnet_load_dataset
Open

modify resnet model to use load dataset#429
ctr-lelanchelianTT wants to merge 2 commits intomainfrom
lelanchelian/check_resnet_load_dataset

Conversation

@ctr-lelanchelianTT
Copy link
Contributor

Ticket

Link to Github Issue

Problem description

Currently, model inputs are handled by links from different sources.This issue aims to standardize input handling across all models by using the Hugging Face load_dataset API as the primary source of input data.

What's changed

modified resnet model to input from load_dataset from HuggingFace

Checklist

  • New/Existing tests provide coverage for changes

I have attached the cpu inference script and results for your reference:
test_cpu_inference.py
check.log

test_resnet.log

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant