Commit 510a6305 authored by Paul Bethge's avatar Paul Bethge
Browse files

update num epochs

parent 0b165738
......@@ -3,11 +3,11 @@
# Copyright 2020 The TensorFlow Authors.
%% Cell type:code id: tags:
```
#@title Licensed under the Apache License, Version 2.0 (the "License");
#@title Licensed under the Apache License, Version 2.0 (the "License"); { display-mode: "form" }
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
......@@ -480,11 +480,11 @@
# Set seeds to get reproducible results
set_seeds()
model_unconstrained = create_model()
model_unconstrained.fit(celeb_a_train_data_wo_group(BATCH_SIZE), epochs=5, steps_per_epoch=1000)
model_unconstrained.fit(celeb_a_train_data_wo_group(BATCH_SIZE), epochs=2, steps_per_epoch=1000)
```
%% Cell type:markdown id: tags:
Evaluating the model on the test data should result in a final accuracy score of just over 90%. Not bad for a simple model with no fine tuning.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment