self.device = device
nb = args.data_size if args.data_size > 0 else 250000
- log_string('generating {nb} samples (can take some time)')
+ log_string(f'generating {nb} samples (can take some time)')
self.train_descr = generate_descr((nb * 4) // 5)
self.test_descr = generate_descr((nb * 1) // 5)