plots_file = os.path.join('plots', save_str)
# relevant variables
num_workers = 4
N0 = 2048
N1 = 1000
N2 = 500
num_classes = len(classes)
init_lr = 0.0001
num_epochs = 3
num_clusters = kmeans.n_clusters
kmeans_dict = kmeans.cluster_centers_
# DATA
# datasets
real_data = MultibinImages(augmented_path, 'real', 'c0', kmeans_file)
render_data = MultibinImages(render_path, 'render', 'c0', kmeans_file)
test_data = Pascal3dAll(pascal3d_path, 'test')
# setup data loaders
real_loader = DataLoader(real_data,
batch_size=4,
shuffle=True,
num_workers=num_workers,
pin_memory=True,
collate_fn=my_collate)
render_loader = DataLoader(render_data,
batch_size=4,
shuffle=True,
num_workers=num_workers,
pin_memory=True,
collate_fn=my_collate)
ndim = 3
num_workers = 4
N0 = 2048
N1 = 1000
N2 = 500
N3 = 100
num_classes = len(classes)
init_lr = 0.0001
num_epochs = 3
problem_type = 'm0'
criterion = loss_m0(1.0)
# DATA
# datasets
real_data = MultibinImages(augmented_path, 'real', problem_type, kmeans_file)
render_data = MultibinImages(render_path, 'render', problem_type, kmeans_file)
test_data = Pascal3dAll(pascal3d_path, 'test')
# setup data loaders
real_loader = DataLoader(real_data,
batch_size=4,
shuffle=True,
num_workers=num_workers,
pin_memory=True,
collate_fn=my_collate)
render_loader = DataLoader(render_data,
batch_size=4,
shuffle=True,
num_workers=num_workers,
pin_memory=True,
collate_fn=my_collate)
log_dir = os.path.join('logs', save_str)
# relevant variables
num_workers = 4
N0 = 2048
N1 = 1000
N2 = 500
num_classes = len(classes)
init_lr = 0.0001
num_epochs = 3
num_clusters = kmeans.n_clusters
kmeans_dict = kmeans.cluster_centers_
# DATA
# datasets
real_data = MultibinImages(pascal3d_path, 'real', 'c0', kmeans_file)
render_data = MultibinImages(pascal3d_path, 'render', 'c0', kmeans_file)
test_data = Pascal3dAll(pascal3d_path, 'test')
# setup data loaders
real_loader = DataLoader(real_data,
batch_size=4,
shuffle=True,
num_workers=num_workers,
pin_memory=True,
collate_fn=my_collate)
render_loader = DataLoader(render_data,
batch_size=4,
shuffle=True,
num_workers=num_workers,
pin_memory=True,
collate_fn=my_collate)