refactor: reorganize project structure by moving core modules and update import paths in API server
This commit is contained in:
@@ -0,0 +1,5 @@
|
||||
from train_cloud_removal import UNet
|
||||
model = UNet(in_channels=6, out_channels=4)
|
||||
print(hasattr(model, 'inc'))
|
||||
print(hasattr(model, 'conv1'))
|
||||
print(list(model.parameters())[0].shape)
|
||||
Reference in New Issue
Block a user