Update Detail_Model_Patch_Custom.py
fixed improper import created by a previous commit
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
zimport torch
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
|
||||
class ModelWrapper(nn.Module):
|
||||
|
||||
Reference in New Issue
Block a user