Commit @2c09074ada4028594f6289daf63117dbe899a7e0

윤영준 2023-06-23

discriminator test code

@2c09074ada4028594f6289daf63117dbe899a7e0

3690a9e

2c09074

model/Discriminator.py

--- model/Discriminator.py

+++ model/Discriminator.py

...	...	@@ -38,3 +38,12 @@
38	38	fc_out = clamp(fc_out, min=1e-7, max=1 - 1e-7)
39	39
40	40	return fc_out, attention_map, fc2
	41	+
	42	+if __name__ == "__main__":
	43	+ import torch
	44	+ from torchinfo import summary
	45	+
	46	+ torch.set_default_tensor_type(torch.FloatTensor)
	47	+ generator = DiscriminativeNet(960,540)
	48	+ batch_size = 2
	49	+ summary(generator, input_size=(batch_size, 3, 960,540))

Add a comment

List