Skip to content

Supervised training #3

Description

@tseckin

I wonder if it is enough to specify "supervised=True" of feeds method for supervised learning?

I set supervised parameter to true for cifar but I am getting an error. (I also tried lfw and getting the same error)

Example for cifar:

    train_feed, test_feed, _ = dataset.cifar.feeds(
        split='test', batch_size=batch_size, epoch_size=epoch_size, supervised=True
    )

Here is the error of cifar_aegan:

INFO     Downloading http://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
INFO     Unpacking datasets/cifar10/cifar-10-python.tar.gz
________________________________________________________________________________
[Memory] Calling dataset.cifar.arrays-alias...
arrays-alias('test')
__________________________________________________arrays-alias - 646.6s, 10.8min
Reconstruction error at layer #6: Convolution
experiment_name: cifar_nhidden256_reconganweight1.0e-05_recondepth6_nodisaerecon
Traceback (most recent call last):
  File "cifar_aegan.py", line 39, in <module>
    run()
  File "cifar_aegan.py", line 29, in run
    model, output_dir, train_feed, test_feed, n_epochs=200, lr_start=0.025
  File "/autoencoding_beyond_pixels/aegan.py", line 80, in train
    margin=gan_margin)
  File "/autoencoding_beyond_pixels/model/aegan.py", line 165, in __init__
    super(GradientDescent, self).__init__(model, feed, learn_rule)
  File "build/bdist.linux-x86_64/egg/deeppy/train/gradient_descent.py", line 18, in __init__
  File "/autoencoding_beyond_pixels/model/aegan.py", line 171, in reset
    self.model.setup(*self.feed.shapes)
TypeError: setup() takes exactly 2 arguments (3 given)

Is it a bug or am I missing something for supervised learning?

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions