How to change number of elements?

[sukritiverma1996]

How to change number of elements?

[ktrk115]

Hi Sukriti,

If I remember correctly, you don't need to edit the code for both the generator and the discriminator. The code should work with variable length of elements. To add more labels, you need to edit the following.

const_layout/data/rico.py

Lines 19 to 33 in 4090f55

	labels = [
	'Toolbar',
	'Image',
	'Text',
	'Icon',
	'Text Button',
	'Input',
	'List Item',
	'Advertisement',
	'Pager Indicator',
	'Web View',
	'Background Image',
	'Drawer',
	'Modal',
	]

If you get any results with the full Rico dataset, I would appreciate it if you could share them here or on PR.

Thanks!
Kotaro

[sukritiverma1996]

Yes, definitely
This is my first time dabbling with GANs though
Thank you!