Update test_model.py
This commit is contained in:
parent
877bf354b9
commit
ecd3a40505
@ -1,6 +1,4 @@
|
||||
import torch
|
||||
from torchsummary import summary
|
||||
import numpy as np
|
||||
|
||||
from nets import Model
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user