6 lines
182 B
Python
6 lines
182 B
Python
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)
|