Skip to content

Training and Benchmarking a Deep Neural Network Using MNIST #167

@AishwaryaM1111

Description

@AishwaryaM1111

i am running Memtorch mnist dataset when i run up till here from memtorch.bh.nonideality.NonIdeality import apply_nonidealities

patched_model_ = apply_nonidealities(copy.deepcopy(patched_model),
non_idealities=[memtorch.bh.nonideality.NonIdeality.NonLinear],
simulate=True)
everything is running fine
but when i run this print(test(patched_model_, test_loader)) its taking too much time like more than 1 hr so is there any solution iam facing when every time i run this code getting same issue. iam using google colab with GPU and i also tried using vs code.
getting this
/content/MemTorch/memtorch/mn/Conv2d.py:247: UserWarning: Non-liner modeling does not currently account for source and line resistances.
warnings.warn(

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions