"rpl": {
"nb_epochs": 40,
"batch_size": 25,
- "nb_train_samples": 1000000,
+ "nb_train_samples": 100000,
"nb_test_samples": 10000,
},
"world": {
nb_train_samples=args.nb_train_samples,
nb_test_samples=args.nb_test_samples,
batch_size=args.batch_size,
+ logger=log_string,
device=device,
)